Jivat Neet Kaur

jnkaur at berkeley dot edu

I am a PhD student at UC Berkeley, where I am happily advised by Ahmed Alaa and Michael I. Jordan. I am affiliated with the Computational Precision Health program and Berkeley Artificial Intelligence Research (BAIR). I am interested in developing principled and robust solutions to ensure the reliable use of machine learning systems in real-world deployments. My research currently builds on tools and concepts from statistics to develop methods for uncertainty quantification and decision-making.

Previously, I was a pre-doctoral research fellow at Microsoft Research, working with Amit Sharma and Emre Kiciman on causal inference and out-of-distribution generalization. I obtained my bachelor’s (B.E.) in Computer Science in 2021 from BITS Pilani, India.

CV |  Google Scholar |  Github |  Twitter |  LinkedIn

Publications

* = equal contribution.

Strategic Feature Selection
Jivat Neet Kaur, Divya Shanmugam, Emma Pierson, Michael I. Jordan, Nika Haghtalab, Ahmed Alaa*, Serena Wang*
Working paper, 2025
Preliminary version to be presented at the NeurIPS Workshops on Reliable ML from Unreliable Data and ACA 2025
Conformal Prediction Sets with Improved Conditional Coverage using Trust Scores
Jivat Neet Kaur, Michael I. Jordan, Ahmed Alaa
arXiv 2025
paper | code
Speaking in Terms of Money: Financial Knowledge Acquisition via Speech Data Generation
Advait Bhat*, Nidhi Kulkarni*, Safiya Husain, Aditya Yadavalli, Jivat Neet Kaur, Anurag Shukla, Monali Shelar, Vivek Seshadri
COMPASS 2024
paper
Modeling the data-generating process is necessary for out-of-distribution generalization
Jivat Neet Kaur, Emre Kiciman, Amit Sharma
ICLR 2023 (spotlight)
ICML Workshop on Spurious Correlations, Invariance and Stability 2022 (spotlight)
paper | code | demo | video
Towards Modular Machine Learning Pipelines
Aditya Modi, Jivat Neet Kaur, Maggie Makar, Pavan Mallapragada, Amit Sharma, Emre Kiciman, Adith Swaminathan
ICML Workshop on Localized Learning (LLW) 2023
paper
LM-CORE: Language models with contextually relevant external knowledge
Jivat Neet Kaur, Sumit Bhatia, Milan Aggarwal, Rachit Bansal, Balaji Krishnamurthy
Findings of NAACL 2022
paper
CoSe-Co: Text Conditioned Generative CommonSense Contextualizer
Rachit Bansal, Milan Aggarwal, Sumit Bhatia, Jivat Neet Kaur, Balaji Krishnamurthy
NAACL 2022
paper
Modern baselines for SPARQL semantic parsing
Debayan Banerjee, Pranav Ajit Nair*, Jivat Neet Kaur*, Ricardo Usbeck, Chris Biemann
SIGIR 2022
paper | code
Ask & Explore: Grounded Question Answering for Curiosity-Driven Exploration
Jivat Neet Kaur, Yiding Jiang, Paul Pu Liang
ICLR Workshop on Embodied Multimodal Learning 2021
paper | code
Simulation and Selection of Detumbling Algorithms for a 3U CubeSat
Vishnu P Katkoori, Jivat Neet Kaur, Tushar Goyal
International Astronautical Congress 2019
paper