Jimmy Ba

According to our database, Jimmy Ba authored at least 78 papers between 2011 and 2024.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of three.

2024

The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearning.

Nathaniel Li, Alexander Pan, Anjali Gopal, Summer Yue, Daniel Berrios, Alice Gatti, Justin D. Li, Ann-Kathrin Dombrowski, Shashwat Goel, Long Phan, Gabriel Mukobi, Nathan Helm-Burger, Rassin Lababidi, Lennart Justen, Andrew B. Liu, Michael Chen, Isabelle Barrass, Oliver Zhang, Xiaoyuan Zhu, Rishub Tamirisa, Bhrugu Bharathi, Adam Khoja, Zhenqi Zhao, Ariel Herbert-Voss, Cort B. Breuer, Andy Zou, Mantas Mazeika, Zifan Wang, Palash Oswal, Weiran Liu, Adam A. Hunt, Justin Tienken-Harder, Kevin Y. Shih, Kemper Talley, John Guan, Russell Kaplan, Ian Steneker, David Campbell, Brad Jokubaitis, Alex Levinson, Jean Wang, William Qian, Kallol Krishna Karmakar, Steven Basart, Stephen Fitz, Mindy Levine, Ponnurangam Kumaraguru, Uday Kiran Tupakula, Vijay Varadharajan, Yan Shoshitaishvili, Jimmy Ba, Kevin M. Esvelt, Alexandr Wang, Dan Hendrycks

CoRR, 2024

2023

Using Large Language Models for Hyperparameter Optimization.

Michael R. Zhang, Nishkrit Desai, Juhan Bae, Jonathan Lorraine, Jimmy Ba

CoRR, 2023

OpenWebMath: An Open Dataset of High-Quality Mathematical Web Text.

Keiran Paster, Marco Dos Santos, Zhangir Azerbayev, Jimmy Ba

CoRR, 2023

Identifying the Risks of LM Agents with an LM-Emulated Sandbox.

Yangjun Ruan, Honghua Dong, Andrew Wang, Silviu Pitis, Yongchao Zhou, Jimmy Ba, Yann Dubois, Chris J. Maddison, Tatsunori Hashimoto

CoRR, 2023

Training on Thin Air: Improve Image Classification with Generated Data.

Yongchao Zhou, Hshmat Sahak, Jimmy Ba

CoRR, 2023

AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedback.

Yann Dubois, Xuechen Li, Rohan Taori, Tianyi Zhang, Ishaan Gulrajani, Jimmy Ba, Carlos Guestrin, Percy Liang, Tatsunori B. Hashimoto

CoRR, 2023

Clinical Camel: An Open-Source Expert-Level Medical Language Model with Dialogue-Based Knowledge Encoding.

Augustin Toma, Patrick R. Lawler, Jimmy Ba, Rahul G. Krishnan, Barry B. Rubin, Bo Wang

CoRR, 2023

Boosted Prompt Ensembles for Large Language Models.

Silviu Pitis, Michael R. Zhang, Andrew Wang, Jimmy Ba

CoRR, 2023

Mastering Diverse Domains through World Models.

Danijar Hafner, Jurgis Pasukonis, Jimmy Ba, Timothy P. Lillicrap

CoRR, 2023

STEVE-1: A Generative Model for Text-to-Behavior in Minecraft.

Shalev Lifshitz, Keiran Paster, Harris Chan, Jimmy Ba, Sheila A. McIlraith

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedback.

Yann Dubois, Chen Xuechen Li, Rohan Taori, Tianyi Zhang, Ishaan Gulrajani, Jimmy Ba, Carlos Guestrin, Percy Liang, Tatsunori B. Hashimoto

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Learning in the Presence of Low-dimensional Structure: A Spiked Random Matrix Perspective.

Jimmy Ba, Murat A. Erdogdu, Taiji Suzuki, Zhichao Wang, Denny Wu

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Classifying Course Discussion Board Questions using LLMs.

Paul Zhang, Brandon Jaipersaud, Jimmy Ba, Andrew Petersen, Lisa Zhang, Michael R. Zhang

Proceedings of the 2023 Conference on Innovation and Technology in Computer Science Education V. 2, 2023

TR0N: Translator Networks for 0-Shot Plug-and-Play Conditional Generation.

Zhaoyan Liu, Noël Vouitsis, Satya Krishna Gorti, Jimmy Ba, Gabriel Loaiza-Ganem

Proceedings of the International Conference on Machine Learning, 2023

Large Language Models are Human-Level Prompt Engineers.

Yongchao Zhou, Andrei Ioan Muresanu, Ziwen Han, Keiran Paster, Silviu Pitis, Harris Chan, Jimmy Ba

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Multi-Rate VAE: Train Once, Get the Full Rate-Distortion Curve.

Juhan Bae, Michael R. Zhang, Michael Ruan, Eric Wang, So Hasegawa, Jimmy Ba, Roger Baker Grosse

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Decomposed Prompting to Answer Questions on a Course Discussion Board.

Brandon Jaipersaud, Paul Zhang, Jimmy Ba, Andrew Petersen, Lisa Zhang, Michael R. Zhang

Proceedings of the Artificial Intelligence in Education. Posters and Late Breaking Results, Workshops and Tutorials, Industry and Innovation Tracks, Practitioners, Doctoral Consortium and Blue Sky, 2023

Residual Prompt Tuning: improving prompt tuning with residual reparameterization.

Anastasia Razdaibiedina, Yuning Mao, Madian Khabsa, Mike Lewis, Rui Hou, Jimmy Ba, Amjad Almahairi

Proceedings of the Findings of the Association for Computational Linguistics: ACL 2023, 2023

2022

Exploring Low Rank Training of Deep Neural Networks.

Siddhartha Rao Kamalakara, Acyr Locatelli, Bharat Venkitesh, Jimmy Ba, Yarin Gal, Aidan N. Gomez

CoRR, 2022

You Can't Count on Luck: Why Decision Transformers Fail in Stochastic Environments.

Keiran Paster, Sheila A. McIlraith, Jimmy Ba

CoRR, 2022

Dataset Distillation using Neural Feature Regression.

Yongchao Zhou, Ehsan Nezhadarya, Jimmy Ba

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

You Can't Count on Luck: Why Decision Transformers and RvS Fail in Stochastic Environments.

Keiran Paster, Sheila A. McIlraith, Jimmy Ba

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representation.

Jimmy Ba, Murat A. Erdogdu, Taiji Suzuki, Zhichao Wang, Denny Wu, Greg Yang

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

Understanding the Variance Collapse of SVGD in High Dimensions.

Jimmy Ba, Murat A. Erdogdu, Marzyeh Ghassemi, Shengyang Sun, Taiji Suzuki, Denny Wu, Tianzong Zhang

Proceedings of the Tenth International Conference on Learning Representations, 2022

2021

Clockwork Variational Autoencoders.

Vaibhav Saxena, Jimmy Ba, Danijar Hafner

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

How does a Neural Network's Architecture Impact its Robustness to Noisy Labels?

Jingling Li, Mozhi Zhang, Keyulu Xu, John Dickerson, Jimmy Ba

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Learning Domain Invariant Representations in Goal-conditioned Block MDPs.

Beining Han, Chongyi Zheng, Harris Chan, Keiran Paster, Michael R. Zhang, Jimmy Ba

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

LIME: Learning Inductive Bias for Primitives of Mathematical Reasoning.

Yuhuai Wu, Markus N. Rabe, Wenda Li, Jimmy Ba, Roger B. Grosse, Christian Szegedy

Proceedings of the 38th International Conference on Machine Learning, 2021

Efficient Statistical Tests: A Neural Tangent Kernel Approach.

Sheng Jia, Ehsan Nezhadarya, Yuhuai Wu, Jimmy Ba

Proceedings of the 38th International Conference on Machine Learning, 2021

INT: An Inequality Benchmark for Evaluating Generalization in Theorem Proving.

Yuhuai Wu, Albert Q. Jiang, Jimmy Ba, Roger Baker Grosse

Proceedings of the 9th International Conference on Learning Representations, 2021

Planning from Pixels using Inverse Dynamics Models.

Keiran Paster, Sheila A. McIlraith, Jimmy Ba

Proceedings of the 9th International Conference on Learning Representations, 2021

Mastering Atari with Discrete World Models.

Danijar Hafner, Timothy P. Lillicrap, Mohammad Norouzi, Jimmy Ba

Proceedings of the 9th International Conference on Learning Representations, 2021

When does preconditioning help or hurt generalization?

Shun-ichi Amari, Jimmy Ba, Roger Baker Grosse, Xuechen Li, Atsushi Nitanda, Taiji Suzuki, Denny Wu, Ji Xu

Proceedings of the 9th International Conference on Learning Representations, 2021

2020

Noisy Labels Can Induce Good Representations.

Jingling Li, Mozhi Zhang, Keyulu Xu, John P. Dickerson, Jimmy Ba

CoRR, 2020

Evaluating Agents without Rewards.

Brendon Matusch, Jimmy Ba, Danijar Hafner

CoRR, 2020

Action and Perception as Divergence Minimization.

Danijar Hafner, Pedro A. Ortega, Jimmy Ba, Thomas Parr, Karl J. Friston, Nicolas Heess

CoRR, 2020

A Study of Gradient Variance in Deep Learning.

Fartash Faghri, David Duvenaud, David J. Fleet, Jimmy Ba

CoRR, 2020

The Scattering Compositional Learner: Discovering Objects, Attributes, Relationships in Analogical Reasoning.

Yuhuai Wu, Honghua Dong, Roger B. Grosse, Jimmy Ba

CoRR, 2020

Learning Intrinsic Rewards as a Bi-Level Optimization Problem.

Bradly C. Stadie, Lunjun Zhang, Jimmy Ba

Proceedings of the Thirty-Sixth Conference on Uncertainty in Artificial Intelligence, 2020

Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning.

Silviu Pitis, Harris Chan, Stephen Zhao, Bradly C. Stadie, Jimmy Ba

Proceedings of the 37th International Conference on Machine Learning, 2020

Improving Transformer Optimization Through Better Initialization.

Xiao Shi Huang, Felipe Pérez, Jimmy Ba, Maksims Volkovs

Proceedings of the 37th International Conference on Machine Learning, 2020

BatchEnsemble: an Alternative Approach to Efficient Ensemble and Lifelong Learning.

Yeming Wen, Dustin Tran, Jimmy Ba

Proceedings of the 8th International Conference on Learning Representations, 2020

On Solving Minimax Optimization Locally: A Follow-the-Ridge Approach.

Yuanhao Wang, Guodong Zhang, Jimmy Ba

Proceedings of the 8th International Conference on Learning Representations, 2020

Exploring Model-based Planning with Policy Networks.

Tingwu Wang, Jimmy Ba

Proceedings of the 8th International Conference on Learning Representations, 2020

An Inductive Bias for Distances: Neural Nets that Respect the Triangle Inequality.

Silviu Pitis, Harris Chan, Kiarash Jamali, Jimmy Ba

Proceedings of the 8th International Conference on Learning Representations, 2020

Dream to Control: Learning Behaviors by Latent Imagination.

Danijar Hafner, Timothy P. Lillicrap, Jimmy Ba, Mohammad Norouzi

Proceedings of the 8th International Conference on Learning Representations, 2020

Generalization of Two-layer Neural Networks: An Asymptotic Viewpoint.

Jimmy Ba, Murat A. Erdogdu, Taiji Suzuki, Denny Wu, Tianzong Zhang

Proceedings of the 8th International Conference on Learning Representations, 2020

An Empirical Study of Stochastic Gradient Descent with Structured Covariance Noise.

Yeming Wen, Kevin Luk, Maxime Gazeau, Guodong Zhang, Harris Chan, Jimmy Ba

Proceedings of the 23rd International Conference on Artificial Intelligence and Statistics, 2020

2019

Benchmarking Model-Based Reinforcement Learning.

Tingwu Wang, Xuchan Bao, Ignasi Clavera, Jerrick Hoang, Yeming Wen, Eric D. Langlois, Shunshi Zhang, Guodong Zhang, Pieter Abbeel, Jimmy Ba

CoRR, 2019

Interplay Between Optimization and Generalization of Stochastic Gradient Descent with Covariance Noise.

Yeming Wen, Kevin Luk, Maxime Gazeau, Guodong Zhang, Harris Chan, Jimmy Ba

CoRR, 2019

ACTRCE: Augmenting Experience via Teacher's Advice For Multi-Goal Reinforcement Learning.

Harris Chan, Yuhuai Wu, Jamie Kiros, Sanja Fidler, Jimmy Ba

CoRR, 2019

Lookahead Optimizer: k steps forward, 1 step back.

Michael R. Zhang, James Lucas, Jimmy Ba, Geoffrey E. Hinton

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

Graph Normalizing Flows.

Jenny Liu, Aviral Kumar, Jimmy Ba, Jamie Kiros, Kevin Swersky

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

Neural Graph Evolution: Towards Efficient Automatic Robot Design.

Tingwu Wang, Yuhao Zhou, Sanja Fidler, Jimmy Ba

Proceedings of the 7th International Conference on Learning Representations, 2019

DOM-Q-NET: Grounded RL on Structured Language.

Sheng Jia, Jamie Kiros, Jimmy Ba

Proceedings of the 7th International Conference on Learning Representations, 2019

2018

Solving Approximate Wasserstein GANs to Stationarity.

Maziar Sanjabi, Jimmy Ba, Meisam Razaviyayn, Jason D. Lee

CoRR, 2018

On the Convergence and Robustness of Training GANs with Regularized Optimal Transport.

Maziar Sanjabi, Jimmy Ba, Meisam Razaviyayn, Jason D. Lee

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 2018

Reversible Recurrent Neural Networks.

Matthew MacKay, Paul Vicol, Jimmy Ba, Roger B. Grosse

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 2018

Flipout: Efficient Pseudo-Independent Weight Perturbations on Mini-Batches.

Yeming Wen, Paul Vicol, Jimmy Ba, Dustin Tran, Roger B. Grosse

Proceedings of the 6th International Conference on Learning Representations, 2018

NerveNet: Learning Structured Policy with Graph Neural Networks.

Tingwu Wang, Renjie Liao, Jimmy Ba, Sanja Fidler

Proceedings of the 6th International Conference on Learning Representations, 2018

Kronecker-factored Curvature Approximations for Recurrent Neural Networks.

James Martens, Jimmy Ba, Matt Johnson

Proceedings of the 6th International Conference on Learning Representations, 2018

2017

Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation

Yuhuai Wu, Elman Mansimov, Roger B. Grosse, Shun Liao, Jimmy Ba

Proceedings of the Advances in Neural Information Processing Systems 30: Annual Conference on Neural Information Processing Systems 2017, 2017

Distributed Second-Order Optimization using Kronecker-Factored Approximations.

Jimmy Ba, Roger B. Grosse, James Martens

Proceedings of the 5th International Conference on Learning Representations, 2017

2016

Actor-Mimic: Deep Multitask and Transfer Reinforcement Learning.

Emilio Parisotto, Lei Jimmy Ba, Ruslan Salakhutdinov

Proceedings of the 4th International Conference on Learning Representations, 2016

Generating Images from Captions with Attention.

Elman Mansimov, Emilio Parisotto, Lei Jimmy Ba, Ruslan Salakhutdinov

Proceedings of the 4th International Conference on Learning Representations, 2016

Layer Normalization.

Lei Jimmy Ba, Jamie Ryan Kiros, Geoffrey E. Hinton

CoRR, 2016

Classifying and segmenting microscopy images with deep multiple instance learning.

Oren Z. Kraus, Lei Jimmy Ba, Brendan J. Frey

Bioinform., 2016

Using Fast Weights to Attend to the Recent Past.

Jimmy Ba, Geoffrey E. Hinton, Volodymyr Mnih, Joel Z. Leibo, Catalin Ionescu

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016

2015

Classifying and Segmenting Microscopy Images Using Convolutional Multiple Instance Learning.

Oren Z. Kraus, Lei Jimmy Ba, Brendan J. Frey

CoRR, 2015

Adam: A Method for Stochastic Optimization.

Diederik P. Kingma, Jimmy Ba

Proceedings of the 3rd International Conference on Learning Representations, 2015

Multiple Object Recognition with Visual Attention.

Jimmy Ba, Volodymyr Mnih, Koray Kavukcuoglu

Proceedings of the 3rd International Conference on Learning Representations, 2015

Learning Wake-Sleep Recurrent Attention Models.

Jimmy Ba, Ruslan Salakhutdinov, Roger B. Grosse, Brendan J. Frey

Proceedings of the Advances in Neural Information Processing Systems 28: Annual Conference on Neural Information Processing Systems 2015, 2015

Show, Attend and Tell: Neural Image Caption Generation with Visual Attention.

Kelvin Xu, Jimmy Ba, Ryan Kiros, Kyunghyun Cho, Aaron C. Courville, Ruslan Salakhutdinov, Richard S. Zemel, Yoshua Bengio

Proceedings of the 32nd International Conference on Machine Learning, 2015

Predicting Deep Zero-Shot Convolutional Neural Networks Using Textual Descriptions.

Lei Jimmy Ba, Kevin Swersky, Sanja Fidler, Ruslan Salakhutdinov

Proceedings of the 2015 IEEE International Conference on Computer Vision, 2015

"SQUEAKeys": A friction idiophone, for physical interaction with mobile devices.

Steve Mann, Ryan E. Janzen, Valmiki Rampersad, Jason Huang, Lei Jimmy Ba

Proceedings of the 2015 IEEE Games Entertainment Media Conference, 2015

2014

Do Deep Nets Really Need to be Deep?

Jimmy Ba, Rich Caruana

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

2013

Adaptive dropout for training deep neural networks.

Lei Jimmy Ba, Brendan J. Frey

Proceedings of the Advances in Neural Information Processing Systems 26: 27th Annual Conference on Neural Information Processing Systems 2013. Proceedings of a meeting held December 5-8, 2013

2011

User-interfaces based on the water-hammer effect: water-hammer piano as an interactive percussion surface.

Steve Mann, Ryan E. Janzen, Jason Huang, Matthew B. Kelly, Lei Jimmy Ba, Alexander Chen

Proceedings of the 5th International Conference on Tangible and Embedded Interaction 2011, 2011