| 2026 | ACL | Bayesian Active Learning with Gaussian Processes Guided by LLM Relevance Scoring for Dense Passage Retrieval. | Junyoung Kim, Anton Korikov, Jiazhou Liang, Justin Cui, Yifan Simon Liu, Qianfeng Wen, Mark Zhao, Scott Sanner |
| 2026 | ACL | Multimodal Item Scoring for Natural Language Recommendation via Gaussian Process Regression with LLM Relevance Judgments. | Yifan Simon Liu, Qianfeng Wen, Jiazhou Liang, Mark Zhao, Justin Cui, Anton Korikov, Armin Toroghi, Junyoung Kim, Scott Sanner |
| 2025 | ICML | OR-Bench: An Over-Refusal Benchmark for Large Language Models. | Justin Cui, Wei-Lin Chiang, Ion Stoica, Cho-Jui Hsieh |
| 2024 | ICML | Ameliorate Spurious Correlations in Dataset Condensation. | Justin Cui, Ruochen Wang, Yuanhao Xiong, Cho-Jui Hsieh |
| 2024 | SIGIR | Retrieval-Augmented Conversational Recommendation with Prompt-based Semi-Structured Natural Language State Tracking. | Sara Kemper, Justin Cui, Kai Dicarlantonio, Kathy Lin, Danjie Tang, Anton Korikov, Scott Sanner |
| 2023 | ICML | Scaling Up Dataset Distillation to ImageNet-1K with Constant Memory. | Justin Cui, Ruochen Wang, Si Si, Cho-Jui Hsieh |