Flow network based generative models for non-iterative diverse candidate generation E Bengio, M Jain, M Korablyov, D Precup, Y Bengio Advances in Neural Information Processing Systems 34, 27381-27394, 2021 | 289 | 2021 |
DROCC: Deep robust one-class classification S Goyal, A Raghunathan, M Jain, HV Simhadri, P Jain International Conference on Machine Learning, 3711-3721, 2020 | 194 | 2020 |
Biological Sequence Design with GFlowNets M Jain, E Bengio, A Hernandez-Garcia, J Rector-Brooks, BFP Dossou, ... International Conference on Machine Learning, 9786-9801, 2022 | 159 | 2022 |
Trajectory balance: Improved credit assignment in gflownets N Malkin, M Jain, E Bengio, C Sun, Y Bengio Advances in Neural Information Processing Systems 35, 5955-5967, 2022 | 136 | 2022 |
Deup: Direct epistemic uncertainty prediction S Lahlou*, M Jain*, H Nekoei, VI Butoi, P Bertin, J Rector-Brooks, ... Transactions on Machine Learning Research, 2023 | 105 | 2023 |
Multi-objective gflownets M Jain, SC Raparthy, A Hernández-Garcıa, J Rector-Brooks, Y Bengio, ... International Conference on Machine Learning, 14631-14653, 2023 | 66 | 2023 |
Learning GFlowNets from partial episodes for improved convergence and stability K Madan, J Rector-Brooks, M Korablyov, E Bengio, M Jain, AC Nica, ... International Conference on Machine Learning, 23467-23483, 2023 | 64 | 2023 |
GFlowNets for AI-driven scientific discovery M Jain, T Deleu, J Hartford, CH Liu, A Hernandez-Garcia, Y Bengio Digital Discovery 2 (3), 557-577, 2023 | 44 | 2023 |
EdgeML: Machine Learning for resource-constrained edge devices DK Dennis, Y Gaurkar, S Gopinath, S Goyal, C Gupta, M Jain, S Jaiswal, ... URL https://github. com/Microsoft/EdgeML, 2020 | 44 | 2020 |
Amortizing intractable inference in large language models EJ Hu*, M Jain*, E Elmoznino, Y Kaddar, G Lajoie, Y Bengio, N Malkin The Twelfth International Conference on Learning Representations, 2024 | 41 | 2024 |
GFlowNet-EM for learning compositional latent variable models EJ Hu, N Malkin, M Jain, KE Everett, A Graikos, Y Bengio International Conference on Machine Learning, 13528-13549, 2023 | 39 | 2023 |
Stochastic generative flow networks L Pan, D Zhang, M Jain, L Huang, Y Bengio Uncertainty in Artificial Intelligence, 1628-1638, 2023 | 23 | 2023 |
Evaluating Generalization in GFlowNets for Molecule Design AC Nica, M Jain, E Bengio, CH Liu, M Korablyov, MM Bronstein, Y Bengio ICLR2022 Machine Learning for Drug Discovery, 2022 | 22 | 2022 |
Thompson sampling for improved exploration in GFlowNets J Rector-Brooks, K Madan, M Jain, M Korablyov, CH Liu, S Chandar, ... arXiv preprint arXiv:2306.17693, 2023 | 19 | 2023 |
Gflowout: Dropout with generative flow networks D Liu, M Jain, BFP Dossou, Q Shen, S Lahlou, A Goyal, N Malkin, ... International Conference on Machine Learning, 21715-21729, 2023 | 16 | 2023 |
PhyloGFN: Phylogenetic inference with generative flow networks M Zhou, Z Yan, E Layne, N Malkin, D Zhang, M Jain, M Blanchette, ... arXiv preprint arXiv:2310.08774, 2023 | 15 | 2023 |
Pre-Training and Fine-Tuning Generative Flow Networks L Pan, M Jain, K Madan, Y Bengio arXiv preprint arXiv:2310.03419, 2023 | 10 | 2023 |
Multi-Fidelity Active Learning with GFlowNets A Hernandez-Garcia, N Saxena, M Jain, CH Liu, Y Bengio arXiv preprint arXiv:2306.11715, 2023 | 10 | 2023 |
Amortizing intractable inference in diffusion models for vision, language, and control S Venkatraman, M Jain, L Scimeca, M Kim, M Sendera, M Hasan, L Rowe, ... arXiv preprint arXiv:2405.20971, 2024 | 8 | 2024 |
Learning diverse attacks on large language models for robust red-teaming and safety tuning S Lee, M Kim, L Cherif, D Dobre, J Lee, SJ Hwang, K Kawaguchi, G Gidel, ... arXiv preprint arXiv:2405.18540, 2024 | 7 | 2024 |