Indian Institute of Technology (Banaras Hindu University)
|AI/ML Research Intern (Hybrid)
Varanasi, Uttar Pradesh, India
Summary
Led research and implementation of continual learning strategies for molecular property prediction, mitigating catastrophic forgetting and balancing model stability with plasticity.
Highlights
Led implementation and co-authored research addressing catastrophic forgetting in molecular property prediction, applying continual learning and refresh-learning strategies to enhance model robustness.
Designed and deployed the MTL-PORL (Multi-Task Learner - Pareto Optimized Refresh Learning) framework using ChemBERTa, integrating refresh learning with Pareto optimization to balance stability and plasticity.
Engineered multi-task/hierarchical gradient aggregation and hyper-gradient based unlearning pipelines, significantly improving knowledge retention across sequential learning episodes.
Conducted extensive experiments on BBBP, bitter, and sweet molecular datasets, achieving Anytime Avg. Accuracies up to 94.89% and Test Accuracies up to 96.86%, while reducing forgetting measures to as low as -0.0063.