I like research that survives contact with reality.

Kabir Jain, Aayam Bansal, Ishaan Gangwani, Ziming Qiu, Raghav Agarwal

A small differentiable projection enforces mass conservation at under 2% overhead and cuts divergence by five orders of magnitude.

Raghav Agarwal, Aayam Bansal, Ishaan Gangwani, Kabir Jain

Found a spectral fingerprint for memorisation inside diffusion models, leading to an early-stopping check more than 10,000 times faster than generating samples.

Jinjie Ni, Yifan Song, Deepanway Ghosal, Bo Li, David Junhao Zhang, Xiang Yue, Fuzhao Xue, Yuntian Deng, Zian Zheng, Kaichen Zhang, Mahir Shah, Kabir Jain, Yang You, Michael Shieh

An any-to-any multimodal benchmark that reaches 0.98 correlation with human-preference arenas.

Jinjie Ni, Fuzhao Xue, Xiang Yue, Yuntian Deng, Mahir Shah, Kabir Jain, Graham Neubig, Yang You

Matched human preference rankings at 0.96 correlation while cutting one evaluation from USD 2,936 to USD 0.60.

Jinjie Ni, Fuzhao Xue, Yuntian Deng, Jason Phang, Kabir Jain, Mahir Shah, Zangwei Zheng, Yang You

An open dataset of naturally occurring user instructions for studying and improving instruction-following models.

Kabir Jain, Mahir Shah, Lee Tze Koon Yoon

A language and graph-based system for identifying relevant law and classifying legal text.

Kabir Jain, Jed Li Xin Lim, Ziming Qiu

A theoretical and experimental study of the Rayleigh disk method for acoustic measurement.

Designed self-powered pressure sensors for real-time medical monitoring, then used AI to optimise them. The project later won Third Grand Prize in Biomedical Engineering at ISEF 2024.

Built a retrieval-augmented system that generated medical reports from chest radiographs. It won Gold at SSEF 2024 and SUTD's Research and Innovation in Healthcare Award.

Used language and graph models to identify relevant law, turning a pro bono workflow into something about 50 times faster. It won the Harry Shum Computer Science Gold Award, first internationally.

Led a smartphone retinal imaging system for diabetic retinopathy screening at the Kutupalong refugee camp. It has screened more than 3,000 people.