Generalization through Simulation: Integrating Simulated and Real Data into Deep Reinforcement Learning for Vision-Based Autonomous Flight. - Katie Kang; Suneel Belkhale; Gregory Kahn; Pieter Abbeel; Sergey Levine (ICRA, 2019)

Hierarchically Integrated Models: Learning to Navigate from Heterogeneous Robots. - Katie Kang; Gregory Kahn; Sergey Levine (CoRL, 2021)

Multi-Robot Deep Reinforcement Learning for Mobile Navigation. - Katie Kang; Gregory Kahn; Sergey Levine (CoRR, 2021)

Lyapunov Density Models: Constraining Distribution Shift in Learning-Based Control. - Katie Kang; Paula Gradu; Jason J. Choi; Michael Janner; Claire J. Tomlin; Sergey Levine (ICML, 2022)

Multi-Task Imitation Learning for Linear Dynamical Systems. - Thomas T. C. K. Zhang; Katie Kang; Bruce D. Lee; Claire J. Tomlin; Sergey Levine; Stephen Tu; Nikolai Matni (L4DC, 2023)

Deep Neural Networks Tend To Extrapolate Predictably. - Katie Kang; Amrith Setlur; Claire J. Tomlin; Sergey Levine (ICLR, 2024)

What Do Learning Dynamics Reveal About Generalization in LLM Reasoning? - Katie Kang; Amrith Setlur; Dibya Ghosh; Jacob Steinhardt; Claire J. Tomlin; Sergey Levine; Aviral Kumar (CoRR, 2024)

Unfamiliar Finetuning Examples Control How Language Models Hallucinate. - Katie Kang; Eric Wallace; Claire J. Tomlin; Aviral Kumar; Sergey Levine (NAACL (Long Papers), 2025)