Ba, Jimmy

Adam: A Method for Stochastic Optimization

Kingma, D. P., & Ba, J. (2014) · arXiv

Stochastic Gradient Optimization TechniquesAdvanced Optimization Algorithms Research

Cited by 98,946

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

Xu, K., Ba, J., Kiros, R., Cho, K., Courville, A., Salakhutdinov, R., Zemel, R., & Bengio, Y. (2015) · arXiv

Multimodal Machine Learning ApplicationsVisual Attention and Saliency Detection

Cited by 8,248

Multiple Object Recognition with Visual Attention

Ba, J., Mnih, V., & Kavukcuoglu, K. (2014) · arXiv

Advanced Neural Network ApplicationsVisual Attention and Saliency Detection

Cited by 808

Dream to Control: Learning Behaviors by Latent Imagination

Hafner, D., Lillicrap, T., Ba, J., & Norouzi, M. (2019) · arXiv

Reinforcement Learning in RoboticsDomain Adaptation and Few-Shot Learning

Cited by 558

Large Language Models Are Human-Level Prompt Engineers

Zhou, Y., Muresanu, A. I., Han, Z., Paster, K., Pitis, S., Chan, H., & Ba, J. (2022) · arXiv

Topic ModelingNatural Language Processing Techniques

Cited by 330

Mastering Atari with Discrete World Models

Hafner, D., Lillicrap, T., Norouzi, M., & Ba, J. (2020) · arXiv

Reinforcement Learning in RoboticsArtificial Intelligence in Games

Cited by 247

Benchmarking Model-Based Reinforcement Learning

Wang, T., Bao, X., Clavera, I., Hoang, J., Wen, Y., Langlois, E., Zhang, S., Zhang, G., Abbeel, P., & Ba, J. (2019) · arXiv

Reinforcement Learning in RoboticsDomain Adaptation and Few-Shot Learning

Cited by 226

Mastering diverse control tasks through world models

Hafner, D., Pasukonis, J., Ba, J., & Lillicrap, T. (2025) · Nature

Reinforcement Learning in RoboticsArtificial Intelligence in Games

Cited by 189

Mastering Diverse Domains through World Models

Hafner, D., Pasukonis, J., Ba, J., & Lillicrap, T. (2023) · arXiv

Reinforcement Learning in RoboticsArtificial Intelligence in Games

Cited by 124

The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearning

Li, N., Pan, A., Gopal, A., Yue, S., Berrios, D., Gatti, A., Li, J. D., Dombrowski, A.-K., Goel, S., Phan, L., et al. (2024) · arXiv

Privacy-Preserving Technologies in DataAdversarial Robustness in Machine Learning

Cited by 90

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

Toma, A., Lawler, P. R., Ba, J., Krishnan, R. G., Rubin, B. B., & Wang, B. (2023) · arXiv

Artificial Intelligence in Healthcare and EducationTopic Modeling

Cited by 36

INT: An Inequality Benchmark for Evaluating Generalization in Theorem Proving

Wu, Y., Jiang, A. Q., Ba, J., & Grosse, R. (2020) · arXiv

AI-based Problem Solving and PlanningLogic, programming, and type systems

Cited by 30

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

Ruan, Y., Dong, H., Wang, A., Pitis, S., Zhou, Y., Ba, J., Dubois, Y., Maddison, C. J., & Hashimoto, T. (2023) · arXiv

Adversarial Robustness in Machine LearningSoftware Testing and Debugging Techniques

Cited by 19

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

Paster, K., Santos, M. D., Azerbayev, Z., & Ba, J. (2023) · arXiv

Topic ModelingMathematics, Computing, and Information Processing

Cited by 4