Hey! I'm a first-year PhD student in Computer Science at Stanford University, advised by Ben Van Roy. I'm broadly interested in machine learning and I am currently focused on two areas: (1) improving credit assignment in reinforcement learning, and (2) developing a new framework for thinking about and evaluating continual learning.
During my Master's in Computer Science at Stanford, I worked on distributional robustness and was advised by Chelsea Finn. Before that, I worked in Andrew Ng's research group, applying deep learning to healthcare. I continued that line of work as a research fellow in Pranav Rajpurkar's group at Harvard Medical School, where I focused on creating multi-institutional datasets for medical imaging (see MAIDA). I grew up in Sweden and finished my undergraduate degree at Chalmers University of Technology.
My main interests outside of AI are playing music with friends and meditation. I'm also interested in education and during college, I co-founded Knowly, an ed-tech company focused on making corporate training more effective. More recently, I worked at Inspirit AI, a company giving live online courses for high school students in artificial intelligence and computer science.
Publications
Extending the WILDS benchmark for unsupervised adaptation
Shiori Sagawa*, Pang Wei Koh*, Tony Lee*, Irena Gao*, Sang Michael Xie, Kendrick Shen, Ananya Kumar, Weihua Hu, Michihiro Yasunaga, Henrik Marklund, Sara Beery, Etienne David, Ian Stavness, Wei Guo, Jure Leskovec, Kate Saenko, Tatsunori Hashimoto, Sergey Levine, Chelsea Finn, and Percy Liang
ICLR, 2022
Wilds: A benchmark of in-the-wild distribution shifts
Pang Wei Koh*, Shiori Sagawa*, Henrik Marklund, Sang Michael Xie, Marvin Zhang, Akshay Balsubramani, Weihua Hu, Michihiro Yasunaga, Richard Lanas Phillips, Irena Gao, Tony Lee, Etienne David, Ian Stavness, Wei Guo, Berton A. Earnshaw, Imran S. Haque, Sara Beery, Jure Leskovec, Anshul Kundaje, Emma Pierson, Sergey Levine, Chelsea Finn, and Percy Liang
ICML, 2021
Adaptive risk minimization: Learning to adapt to domain shift
Marvin Zhang*, Henrik Marklund*, Nikita Dhawan*, Abhishek Gupta, Sergey Levine, and Chelsea Finn
Advances in Neural Information Processing Systems, 2021
Chexpert: A large chest radiograph dataset with uncertainty labels and expert comparison
Jeremy Irvin*, Pranav Rajpurkar*, Michael Ko, Yifan Yu, Silviana Ciurea-Ilcus, Chris Chute, Henrik Marklund, Behzad Haghgoo, Robyn Ball, Katie Shpanskaya, Jayne Seekins, David A. Mong, Safwan S. Halabi, Jesse K. Sandberg, Ricky Jones, David B. Larson, Curtis P. Langlotz, Bhavik N. Patel, Matthew P. Lungren, Andrew Y. Ng
Proceedings of the AAAI conference on artificial intelligence, 2019
Deep learning assistance for the histopathologic diagnosis of Helicobacter pylori
Sharon Zhou*, Henrik Marklund*, Ondrej Blaha, Manisha Desai, Brock Martin, David Bingham, Gerald J. Berry, Ellen Gomulia, Andrew Y. Ng, Jeanne Shen
Intelligence-Based Medicine, 2020