Duo-LLM: A Framework for Studying Adaptive Computation in Large Language Models. - Keivan Alizadeh; Iman Mirzadeh; Hooman Shahrokhi; Dmitry Belenko; Frank Sun; Minsik Cho; Mohammad Hossein Sekhavat; Moin Nabi; Mehrdad Farajtabar (CoRR, 2024)

GSM-Symbolic: Understanding the Limitations of Mathematical Reasoning in Large Language Models. - Iman Mirzadeh; Keivan Alizadeh; Hooman Shahrokhi; Oncel Tuzel; Samy Bengio; Mehrdad Farajtabar (ICLR, 2025)

Conformal Prediction Sets for Deep Generative Models via Reduction to Conformal Regression. - Hooman Shahrokhi; Devjeet Raj Roy; Yan Yan 006; Venera Arnaoudova; Jana Doppa (UAI, 2025)

Direct Prediction Set Minimization via Bilevel Conformal Classifier Training. - Yuanjie Shi; Hooman Shahrokhi; Xuesong Jia; Xiongzhi Chen; Janardhan Rao Doppa; Yan Yan 006 (CoRR, 2025)