Emergence of Pragmatics from Referential Game between Theory of Mind Agents. - Luyao Yuan; Zipeng Fu; Jingyue Shen; Lu Xu; Junhong Shen; Song-Chun Zhu (CoRR, 2020)

Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation. - Junhong Shen; Lin F. Yang (AAAI, 2021)

AutoML Decathlon: Diverse Tasks, Modern Methods, and Efficiency at Scale. - Nicholas Roberts; Samuel Guo; Cong Xu; Ameet Talwalkar; David Lander; Lvfang Tao; Linhang Cai; Shuaicheng Niu; Jianyu Heng; Hongyang Qin; Minwen Deng; Johannes Hog; et al. (NeurIPS (Competition and Demos), 2021)

Iterative Teacher-Aware Learning. - Luyao Yuan; Dongruo Zhou; Junhong Shen; Jingdong Gao; Jeffrey L. Chen; Quanquan Gu; Ying Nian Wu; Song-Chun Zhu (NeurIPS, 2021)

Mathematical Reconstruction of Patient-Specific Vascular Networks Based on Clinical Images and Global Optimization. - Junhong Shen; Abdul Hannan Faruqi; Yifan Jiang; Nima Maftoon (IEEE Access, 2021)

Efficient Architecture Search for Diverse Tasks. - Junhong Shen; Mikhail Khodak; Ameet Talwalkar (NeurIPS, 2022)

NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks. - Renbo Tu; Nicholas Roberts; Mikhail Khodak; Junhong Shen; Frederic Sala; Ameet Talwalkar (NeurIPS, 2022)

Cross-Modal Fine-Tuning: Align then Refine. - Junhong Shen; Liam Li; Lucio M. Dery; Corey Staten; Mikhail Khodak; Graham Neubig; Ameet Talwalkar (ICML, 2023)

Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains. - Junhong Shen; Neil A. Tenenholtz; James Brian Hall; David Alvarez-Melis; Nicolò Fusi (ICML, 2024)

UPS: Towards Foundation Models for PDE Solving via Cross-Modal Adaptation. - Junhong Shen; Tanya Marwah; Ameet Talwalkar (CoRR, 2024)

ScribeAgent: Towards Specialized Web Agents Using Production-Scale Workflow Data. - Junhong Shen; Atishay Jain; Zedian Xiao; Ishan Amlekar; Mouad Hadji; Aaron Podolny; Ameet Talwalkar (CoRR, 2024)

UPS: Efficiently Building Foundation Models for PDE Solving via Cross-Modal Adaptation. - Junhong Shen; Tanya Marwah; Ameet Talwalkar (Trans. Mach. Learn. Res., 2024)

Specialized Foundation Models Struggle to Beat Supervised Baselines. - Zongzhe Xu; Ritvik Gupta; Wenduo Cheng; Alexander Shen 003; Junhong Shen; Ameet Talwalkar; Mikhail Khodak (ICLR, 2025)

L2G: Repurposing Language Models for Genomics Tasks. - Wenduo Cheng; Junhong Shen; Mikhail Khodak; Jian Ma; Ameet Talwalkar (Trans. Mach. Learn. Res., 2025)

CAT: Content-Adaptive Image Tokenization. - Junhong Shen; Kushal Tirumala; Michihiro Yasunaga; Ishan Misra; Luke Zettlemoyer; Lili Yu; Chunting Zhou (CoRR, 2025)

Mixture-of-Mamba: Enhancing Multi-Modal State-Space Models with Modality-Aware Sparsity. - Weixin Liang; Junhong Shen; Genghan Zhang; Ning Dong; Luke Zettlemoyer; Lili Yu (CoRR, 2025)

CodePDE: An Inference Framework for LLM-driven PDE Solver Generation. - Shanda Li; Tanya Marwah; Junhong Shen; Weiwei Sun; Andrej Risteski; Yiming Yang; Ameet Talwalkar (CoRR, 2025)

Thinking vs. Doing: Agents that Reason by Scaling Test-Time Interaction. - Junhong Shen; Hao Bai; Lunjun Zhang; Yifei Zhou; Amrith Setlur; Shengbang Tong; Diego Caples; Nan Jiang 008; Tong Zhang 001; Ameet Talwalkar; Aviral Kumar (CoRR, 2025)