Jimmy Lei Ba

Adam: A Method for Stochastic Optimization
Diederik P. Kingma, Jimmy Lei Ba
2014arXiv (Cornell University)99K citations Open access
OpenAlex
Show, Attend and Tell: Neural Image Caption Generation with Visual Attention
Kelvin Xu, Jimmy Lei Ba, Ryan Kiros, Kyunghyun Cho +4 more
2015arXiv (Cornell University)8,248 citations Open access
OpenAlex
Layer Normalization
Jimmy Lei Ba, Jamie Kiros, Geoffrey E. Hinton
2016arXiv (Cornell University)3,266 citations Open access
OpenAlex
Do Deep Nets Really Need to be Deep?
Jimmy Lei Ba, Rich Caruana
2013arXiv (Cornell University)1,533 citations Open access
OpenAlex
Multiple Object Recognition with Visual Attention
Jimmy Lei Ba, Volodymyr Mnih, Koray Kavukcuoglu
2014arXiv (Cornell University)808 citations Open access
OpenAlex
Dream to Control: Learning Behaviors by Latent Imagination
Danijar Hafner, Timothy Lillicrap, Jimmy Lei Ba, Mohammad Norouzi
2019arXiv (Cornell University)531 citations Open access
OpenAlex
Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation
Yuhuai Wu, Elman Mansimov, Liao, Shun, Roger Grosse +1 more
2017arXiv (Cornell University)477 citations Open access
OpenAlex
Classifying and segmenting microscopy images with deep multiple instance learning
Oren Z. Kraus, Jimmy Lei Ba, Brendan J. Frey
2016Bioinformatics411 citations Open access
OpenAlex
Lookahead Optimizer: k steps forward, 1 step back
Michael R. Zhang, James Lucas, Geoffrey E. Hinton, Jimmy Lei Ba
2019arXiv (Cornell University)345 citations Open access
OpenAlex
Generating Images from Captions with Attention
Elman Mansimov, Emilio Parisotto, Jimmy Lei Ba, Ruslan Salakhutdinov
2015arXiv (Cornell University)335 citations Open access
OpenAlex
Predicting Deep Zero-Shot Convolutional Neural Networks Using Textual Descriptions
Jimmy Lei Ba, Kevin Swersky, Sanja Fidler, Ruslan Salakhutdinov
2015IEEE International Conference on Computer Vision (ICCV)334 citations
OpenAlex
Large Language Models Are Human-Level Prompt Engineers
Yongchao Zhou, Andrei Ioan Muresanu, Ziwen Han, Keiran Paster +3 more
2022arXiv (Cornell University)330 citations Open access
OpenAlex
Actor-Mimic: Deep Multitask and Transfer Reinforcement Learning
Emilio Parisotto, Jimmy Lei Ba, Ruslan Salakhutdinov
2015arXiv (Cornell University)325 citations Open access
OpenAlex
Adaptive dropout for training deep neural networks
Jimmy Lei Ba, Brendan J. Frey
2013neural information processing systems288 citations
OpenAlex
Automated analysis of high‐content microscopy data with deep learning
Oren Z. Kraus, Ben T. Grys, Jimmy Lei Ba, Yolanda T. Chong +3 more
2017Molecular Systems Biology272 citations Open access
OpenAlex
Mastering Atari with Discrete World Models
Danijar Hafner, Timothy Lillicrap, Mohammad Norouzi, Jimmy Lei Ba
2020arXiv (Cornell University)247 citations Open access
OpenAlex
BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learning
Yeming Wen, Dustin Tran, Jimmy Lei Ba
2020arXiv (Cornell University)233 citations Open access
OpenAlex
Benchmarking Model-Based Reinforcement Learning
Tingwu Wang, Xuchan Bao, Ignasi Clavera, Jerrick Hoang +6 more
2019arXiv (Cornell University)227 citations Open access
OpenAlex
Flipout: Efficient Pseudo-Independent Weight Perturbations on Mini-Batches
Yeming Wen, Paul A. Vicol, Jimmy Lei Ba, Dustin Tran +1 more
2018arXiv (Cornell University)208 citations Open access
OpenAlex
Mastering diverse control tasks through world models
Danijar Hafner, Jurgis Pašukonis, Jimmy Lei Ba, Timothy Lillicrap
2025Nature178 citations Open access
OpenAlex
Lookahead Optimizer: k steps forward, 1 step back
Michael R. Zhang, James Lucas, Jimmy Lei Ba, Geoffrey E. Hinton
2019neural information processing systems154 citations
OpenAlex
NerveNet: Learning Structured Policy with Graph Neural Networks
Tingwu Wang, Renjie Liao, Jimmy Lei Ba, Sanja Fidler
2018International Conference on Learning Representations150 citations
OpenAlex
Using Fast Weights to Attend to the Recent Past
Jimmy Lei Ba, Geoffrey E. Hinton, Volodymyr Mnih, Joel Zaidspiner Leibo +1 more
2016arXiv (Cornell University)146 citations Open access
OpenAlex
Mastering Diverse Domains through World Models
Danijar Hafner, Jurgis Pašukonis, Jimmy Lei Ba, Timothy Lillicrap
2023arXiv (Cornell University)124 citations Open access
OpenAlex
Predicting Deep Zero-Shot Convolutional Neural Networks using Textual Descriptions
Jimmy Lei Ba, Kevin Swersky, Sanja Fidler, Ruslan Salakhutdinov
2015arXiv (Cornell University)117 citations Open access
OpenAlex
Exploring Model-based Planning with Policy Networks
Tingwu Wang, Jimmy Lei Ba
2019arXiv (Cornell University)85 citations Open access
OpenAlex
Graph Normalizing Flows
Liu, Jenny, Aviral Kumar, Jimmy Lei Ba, Jamie Kiros +1 more
2019arXiv (Cornell University)77 citations Open access
OpenAlex
Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation
Yuhuai Wu, Elman Mansimov, Roger Grosse, S. Matthew Liao +1 more
2017neural information processing systems76 citations
OpenAlex
On the Convergence and Robustness of Training GANs with Regularized Optimal Transport
Maziar Sanjabi, Jimmy Lei Ba, Meisam Razaviyayn, Jason D. Lee
2018arXiv (Cornell University)67 citations Open access
OpenAlex
Distributed Second-Order Optimization using Kronecker-Factored Approximations
Jimmy Lei Ba, Roger Grosse, James Martens
2017International Conference on Learning Representations62 citations
OpenAlex
Improving Transformer Optimization Through Better Initialization
Xiao Shi Huang, Juan Manuel Pérez, Jimmy Lei Ba, Maksims Volkovs
2020International Conference on Machine Learning57 citations
OpenAlex
Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning
Silviu Pitis, Harris Chan, Stephen Zhao, Bradly C. Stadie +1 more
2020arXiv (Cornell University)52 citations Open access
OpenAlex
On Solving Minimax Optimization Locally: A Follow-the-Ridge Approach
Yuanhao Wang, Guodong Zhang, Jimmy Lei Ba
2019arXiv (Cornell University)52 citations Open access
OpenAlex
Kronecker-factored Curvature Approximations for Recurrent Neural Networks
James Martens, Jimmy Lei Ba, Matt R. Johnson
2018International Conference on Learning Representations46 citations
OpenAlex
Neural Graph Evolution: Towards Efficient Automatic Robot Design
Tingwu Wang, Yuhao Zhou, Sanja Fidler, Jimmy Lei Ba
2019arXiv (Cornell University)38 citations Open access
OpenAlex
Reversible Recurrent Neural Networks
Matthew Mackay, Paul A. Vicol, Jimmy Lei Ba, Roger Grosse
2018arXiv (Cornell University)38 citations Open access
OpenAlex
Learning Wake-Sleep Recurrent Attention Models
Jimmy Lei Ba, Roger Grosse, Ruslan Salakhutdinov, Brendan J. Frey
2015arXiv (Cornell University)38 citations Open access
OpenAlex
Clinical Camel: An Open Expert-Level Medical Language Model with Dialogue-Based Knowledge Encoding
Augustin Toma, Patrick R. Lawler, Jimmy Lei Ba, Rahul G. Krishnan +2 more
2023arXiv (Cornell University)36 citations Open access
OpenAlex
Generalization of Two-layer Neural Networks: An Asymptotic Viewpoint
Jimmy Lei Ba, Murat A. Erdogdu, Taiji Suzuki, Denny C.-Y. Wu +1 more
2020International Conference on Learning Representations33 citations
OpenAlex
Action and Perception as Divergence Minimization
Danijar Hafner, Pedro A. Ortega, Jimmy Lei Ba, Thomas Parr +2 more
2020arXiv (Cornell University)30 citations Open access
OpenAlex
The Scattering Compositional Learner: Discovering Objects, Attributes, Relationships in Analogical Reasoning
Yuhuai Wu, Honghua Dong, Roger Grosse, Jimmy Lei Ba
2020arXiv (Cornell University)28 citations Open access
OpenAlex
Residual Prompt Tuning: improving prompt tuning with residual reparameterization
Anastasia Razdaibiedina, Yuning Mao, Madian Khabsa, Mike Lewis +3 more
202325 citations Open access
OpenAlex
ACTRCE: Augmenting Experience via Teacher's Advice For Multi-Goal Reinforcement Learning
Harris Chan, Yuhuai Wu, Jamie Kiros, Sanja Fidler +1 more
2019arXiv (Cornell University)20 citations Open access
OpenAlex
Clockwork Variational Autoencoders
Vaibhav Saxena, Jimmy Lei Ba, Danijar Hafner
2021arXiv (Cornell University)18 citations Open access
OpenAlex
Interplay Between Optimization and Generalization of Stochastic Gradient Descent with Covariance Noise.
Yeming Wen, Kevin Luk, Maxime Gazeau, Guodong Zhang +2 more
2019arXiv (Cornell University)18 citations Open access
OpenAlex
The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearning
Nathaniel Li, Pan, Alexander, Anjali Gopal, Summer Yue +53 more
2024arXiv (Cornell University)17 citations Open access
OpenAlex
Identifying the Risks of LM Agents with an LM-Emulated Sandbox
Yangjun Ruan, Honghua Dong, Andrew Wang, Silviu Pitis +5 more
2023arXiv (Cornell University)17 citations Open access
OpenAlex
PIFiA: self-supervised approach for protein functional annotation from single-cell imaging data
Anastasia Razdaibiedina, Alexander V. Brechalov, Helena Friesen, Mojca Mattiazzi Ušaj +6 more
2024Molecular Systems Biology15 citations Open access
OpenAlex
Boosted Prompt Ensembles for Large Language Models
Silviu Pitis, Michael R. Zhang, Andrew Wang, Jimmy Lei Ba
2023arXiv (Cornell University)15 citations Open access
OpenAlex
LIME: Learning Inductive Bias for Primitives of Mathematical Reasoning
Yuhuai Wu, Markus N. Rabe, Wenda Li, Jimmy Lei Ba +2 more
2021arXiv (Cornell University)14 citations Open access
OpenAlex

Most-cited papers