About Me

I expect to receive my Master of Science in Engineering in Computer Graphics and Game Technology from the University of Pennsylvania in May 2025. Here I am supervised by Professor Lingjie Liu. Before that, I received Bachelor of Science in Computer Science (with Honors) and Mathematics from the University of Wisconsin-Madison, where I worked in Machine Learning and Optimization Theory (MLOPT) Research Group and advised by Professor Ramya Korlakai Vinayak, Professor Matthew L. Malloy, and Professor Steven J. Schrodi.

Research interests

Safe and Robust ML: Safety and robustness entail the model’s resilience against adversarial attacks, unanticipated scenarios, and data inconsistencies. My research endeavors include methods such as uncertainty estimation, calibration, and Out-of-Distribution detection to enhance the safety and robustness of ML systems. Relevant publications are listed below:

• Threshold-Based Auto-Labeling: [NeurIPS’23]
• Human-in-the-Loop Out-of-Distribution Detection: [ICML’22], [AISTATS’24]
• Confidence Regions for Multinomial Parameters: [ISIT’22]

AI in Biomedical Research: I am passionate about using AI to address biomedical and healthcare challenges. Check out our ongoing research on [Preprint’21] which looks into the interpretability of neural networks in predicting diabetes in mice. My most recent work involves developing a high-throughput ML system for malaria detection during my internship at Cephla.

Quantifying Uncertainty in Neural Rendering: Neural rendering often leaves questions about the reliability of its reconstructions. Incorporating uncertainty estimation ensures that the results are quantifiably trustworthy. My focus is on infusing Bayesian statistics into neural rendering. Check out our latest work in Gaussian Splatting [Preprint’24].

Publications

Taming False Positives in Out-of-Distribution Detection with Human Feedback
Harit Vishwakarma, Heguang Lin, Ramya Korlakai Vinayak
Artificial Intelligence and Statistics (AISTATS), 2024
[arXiv]

Promises and Pitfalls of Threshold-based Auto-labeling
Harit Vishwakarma, Heguang Lin, Frederic Sala, Ramya Korlakai Vinayak
Neural Information Processing Systems (NeurIPS), 2023 (Spotlight)
[Project Page] [arXiv]

Geometry of the Minimum Volume Confidence Sets
Heguang Lin, Mengze Li, Daniel Pimentel-Alarcón, Matthew L. Malloy
IEEE International Symposium on Information Theory (ISIT), 2022.
[arXiv] [Video]

Adaptive Out-of-Distribution Detection with Human-in-the-Loop
Heguang Lin*, Harit Vishwakarma*, Ramya Korlakai Vinayak
International Conference on Machine Learning (ICML), Workshop on Human-Machine Collaboration and Teaming 2022.
[PDF] [Video]

Preprints

Octopi: Open configurable high-throughput imaging platform for disease diagnosis
Hongquan Li, Heguang Lin, Rinni Bhansali, Pranav Shrestha, You Yan, Kevin Marx, Wei Ouyang, Lucas Fuentes Valenzuela, Ethan Li, Anesta Kothari, Jerome Nowak, Hazel Soto-Montoya, Adil Jussupov, Maxime Voisin, Byaruhanga Oswald, Prasanna Jagannathan, Manu Prakash
In submission, 2024.
[Coming Soon]

HDGS: Textured 2D Gaussian Splatting for Enhanced Scene Rendering
Yunzhou Song*, Heguang Lin*, Jiahui Lei, Lingjie Liu, Kostas Daniilidis
In submission, 2024.
[Project Page] [arXiv]

Machine Learning for Glucose Prediction to Identify Diabetes-related Metabolic Pathways
Gautam Agarwal*, Collin Frink*, Brain Hu*, Ziling Hu*, Eliot Kim*,Heguang Lin*
ProjectX Undergraduate Machine Learning Research Competition, 2021.
[OpenReview]

* Denotes equal contribution.

Working experience

Cephla

• Machine Learning Research Intern
• January 2024 - Present

Genentech

• AI Intern
• June 2020 - August 2020

For fun

Fortune-Telling Chatbot: Check out our Fortune-Telling GPTs on

• Dream Interpretation (周公解梦)
• Lenormand (雷诺曼)
• Master Namewright (取名大师)

The Beta version works best with Chinese, but feel free to tweak the prompt to talk in any other language!

3D Modeling: Just started my journey to modeling and Maya. Check out some of my works HERE!

Awards

The David Dewitt Undergraduate Scholarship, Computer Science Department, UW–Madison, 2022
Undergraduate Scholarship for Summer Study, UW–Madison, 2020