Publications

(* indicates equal contribution)

• Robust Multi-Task Learning with Excess Risks.
  Yifei He, Shiji Zhou, Guojun Zhang, Hyokun Yun, Yi Xu, Belinda Zeng, Trishul Chilimbi, Han Zhao.
  Under review.
• Gradual Domain Adaptation: Theory and Algorithms.[code]
  Yifei He*, Haoxiang Wang*, Bo Li, Han Zhao.
  Under review.
• Efficient Modality Selection in Multimodal Learning.
  Yifei He*, Runxiang Cheng*, Gargi Balasubramaniam*, Yao-Hung Hubert Tsai, Han Zhao.
  In Journal of Machine Learning Research. (JMLR 2024)
  (Extended version of our UAI 2022 paper “Greedy Modality Selection via Approximate Submodular Maximization.”)
• Greedy Modality Selection via Approximate Submodular Maximization.
  Runxiang Cheng*, Gargi Balasubramaniam*, Yifei He*, Yao-Hung Hubert Tsai, Han Zhao.
  In Proceedings of the 38th Conference on Uncertainty in Artificial Intelligence. (UAI 2022)
• A Hierarchical Approach to Multi-Event Survival Analysis.
  Donna Tjandra, Yifei He, Jenna Wiens.
  In Proceedings of the 35th AAAI Conference on Artificial Intelligence. (AAAI 2021)
• Conformer-RL: A Deep Reinforcement Learning Library for Conformer Generation.
  Runxuan Jiang, Tarun Gogineni, Joshua Kammeraad, Yifei He, Ambuj Tewari, Paul Zimmerman.
  In Journal of Computational Chemistry.