Beirami, Ahmad

38 publications

ICLR 2025 Block Verification Accelerates Speculative Decoding Ziteng Sun, Uri Mendlovic, Yaniv Leviathan, Asaf Aharoni, Jae Hun Ro, Ahmad Beirami, Ananda Theertha Suresh

TMLR 2025 CoDe: Blockwise Control for Denoising Diffusion Models Anuj Singh, Sayak Mukherjee, Ahmad Beirami, Hadi J. Rad

ICLRW 2025 CoDe: Blockwise Control for Denoising Diffusion Models Anuj Singh, Sayak Mukherjee, Ahmad Beirami, Hadi J. Rad

AISTATS 2025 Fundamental Limits of Perfect Concept Erasure Somnath Basu Roy Chowdhury, Kumar Avinava Dubey, Ahmad Beirami, Rahul Kidambi, Nicholas Monath, Amr Ahmed, Snigdha Chaturvedi

ICML 2025 Generalization and Robustness of the Tilted Empirical Risk Gholamali Aminian, Amir R. Asadi, Tian Li, Ahmad Beirami, Gesine Reinert, Samuel N. Cohen

CVPR 2025 Immune: Improving Safety Against Jailbreaks in Multi-Modal LLMs via Inference-Time Alignment Soumya Suvra Ghosal, Souradip Chakraborty, Vaibhav Singh, Tianrui Guan, Mengdi Wang, Ahmad Beirami, Furong Huang, Alvaro Velasquez, Dinesh Manocha, Amrit Singh Bedi

ICLRW 2025 Inducing Group Fairness in Prompt-Based Language Model Decisions James Atwood, Nino Scherrer, Preethi Lahoti, Ananth Balashankar, Flavien Prost, Ahmad Beirami

ICML 2025 InfAlign: Inference-Aware Language Model Alignment Ananth Balashankar, Ziteng Sun, Jonathan Berant, Jacob Eisenstein, Michael Collins, Adrian Hutter, Jong Lee, Chirag Nagpal, Flavien Prost, Aradhana Sinha, Ananda Theertha Suresh, Ahmad Beirami

ICLR 2025 Mitigating Object Hallucination in MLLMs via Data-Augmented Phrase-Level Alignment Pritam Sarkar, Sayna Ebrahimi, Ali Etemad, Ahmad Beirami, Sercan O Arik, Tomas Pfister

TMLR 2025 Robust Preference Optimization Through Reward Model Distillation Adam Fisch, Jacob Eisenstein, Vicky Zayats, Alekh Agarwal, Ahmad Beirami, Chirag Nagpal, Peter Shaw, Jonathan Berant

ICLR 2025 Safety Alignment Should Be Made More than Just a Few Tokens Deep Xiangyu Qi, Ashwinee Panda, Kaifeng Lyu, Xiao Ma, Subhrajit Roy, Ahmad Beirami, Prateek Mittal, Peter Henderson

ICML 2025 Theoretical Guarantees on the Best-of-N Alignment Policy Ahmad Beirami, Alekh Agarwal, Jonathan Berant, Alexander D’Amour, Jacob Eisenstein, Chirag Nagpal, Ananda Theertha Suresh

ICMLW 2024 Block Verification Accelerates Speculative Decoding Ziteng Sun, Uri Mendlovic, Yaniv Leviathan, Asaf Aharoni, Ahmad Beirami, Jae Hun Ro, Ananda Theertha Suresh

TMLR 2024 Break It, Imitate It, Fix It: Robustness by Generating Human-like Attacks Aradhana Sinha, Ananth Balashankar, Ahmad Beirami, Thi Avrahami, Jilin Chen, Alex Beutel

ICLRW 2024 Break It, Imitate It, Fix It: Robustness by Generating Human-like Attacks Aradhana Sinha, Ananth Balashankar, Ahmad Beirami, Thi Avrahami, Jilin Chen, Alex Beutel

ICML 2024 Controlled Decoding from Language Models Sidharth Mudgal, Jong Lee, Harish Ganapathy, Yaguang Li, Tao Wang, Yanping Huang, Zhifeng Chen, Heng-Tze Cheng, Michael Collins, Trevor Strohman, Jilin Chen, Alex Beutel, Ahmad Beirami

ICLR 2024 Enhancing Group Fairness in Online Settings Using Oblique Decision Forests Somnath Basu Roy Chowdhury, Nicholas Monath, Ahmad Beirami, Rahul Kidambi, Kumar Avinava Dubey, Amr Ahmed, Snigdha Chaturvedi

ICML 2024 FRAPPÉ: A Group Fairness Framework for Post-Processing Everything Alexandru Tifrea, Preethi Lahoti, Ben Packer, Yoni Halpern, Ahmad Beirami, Flavien Prost

AISTATS 2024 Improving Robustness via Tilted Exponential Layer: A Communication-Theoretic Perspective Bhagyashree Puranik, Ahmad Beirami, Yao Qin, Upamanyu Madhow

ICMLW 2024 Reuse Your Rewards: Reward Model Transfer for Zero-Shot Cross-Lingual Alignment Zhaofeng Wu, Ananth Balashankar, Yoon Kim, Jacob Eisenstein, Ahmad Beirami

NeurIPSW 2023 Controlled Decoding from Language Models Sidharth Mudgal, Jong Lee, Harish Ganapathy, YaGuang Li, Tao Wang, Yanping Huang, Zhifeng Chen, Heng-Tze Cheng, Michael Collins, Jilin Chen, Alex Beutel, Ahmad Beirami

JMLR 2023 On Tilted Losses in Machine Learning: Theory and Applications Tian Li, Ahmad Beirami, Maziar Sanjabi, Virginia Smith

NeurIPSW 2023 Reward Model Underspecification in Language Model Alignment Jacob Eisenstein, Jonathan Berant, Chirag Nagpal, Alekh Agarwal, Ahmad Beirami, Alexander Nicholas D'Amour, Krishnamurthy Dj Dvijotham, Katherine A Heller, Stephen Robert Pfohl, Deepak Ramachandran

ICMLW 2023 Robust Deep Learning via Layerwise Tilted Exponentials Bhagyashree Puranik, Ahmad Beirami, Yao Qin, Upamanyu Madhow

TMLR 2023 Robustness Through Data Augmentation Loss Consistency Tianjian Huang, Shaunak Ashish Halbe, Chinnadhurai Sankar, Pooyan Amini, Satwik Kottur, Alborz Geramifard, Meisam Razaviyayn, Ahmad Beirami

ICMLW 2023 Robustness Through Data Augmentation Loss Consistency Tianjian Huang, Shaunak Halbe, Chinnadhurai Sankar, Pooyan Amini, Satwik Kottur, Alborz Geramifard, Meisam Razaviyayn, Ahmad Beirami

NeurIPSW 2023 SpecTr++: Improved Transport Plans for Speculative Decoding of Large Language Models Kwangjun Ahn, Ahmad Beirami, Ziteng Sun, Ananda Theertha Suresh

NeurIPS 2023 SpecTr: Fast Speculative Decoding via Optimal Transport Ziteng Sun, Ananda Theertha Suresh, Jae Hun Ro, Ahmad Beirami, Himanshu Jain, Felix Yu

ICMLW 2023 SpecTr: Fast Speculative Decoding via Optimal Transport Ziteng Sun, Ananda Theertha Suresh, Jae Hun Ro, Ahmad Beirami, Himanshu Jain, Felix Yu, Michael Riley, Sanjiv Kumar

NeurIPS 2023 Uncovering the Hidden Dynamics of Video Self-Supervised Learning Under Distribution Shifts Pritam Sarkar, Ahmad Beirami, Ali Etemad

TMLR 2022 A Stochastic Optimization Framework for Fair Risk Minimization Andrew Lowy, Sina Baharlouei, Rakesh Pavan, Meisam Razaviyayn, Ahmad Beirami

NeurIPSW 2022 A Stochastic Optimization Framework for Fair Risk Minimization Andrew Lowy, Sina Baharlouei, Rakesh Pavan, Meisam Razaviyayn, Ahmad Beirami

ICML 2021 Ditto: Fair and Robust Federated Learning Through Personalization Tian Li, Shengyuan Hu, Ahmad Beirami, Virginia Smith

ICLR 2021 Tilted Empirical Risk Minimization Tian Li, Ahmad Beirami, Maziar Sanjabi, Virginia Smith

ICLR 2020 Fair Resource Allocation in Federated Learning Tian Li, Maziar Sanjabi, Ahmad Beirami, Virginia Smith

ICLR 2020 Rényi Fair Inference Sina Baharlouei, Maher Nouiehed, Ahmad Beirami, Meisam Razaviyayn

AAAI 2018 On Data-Dependent Random Features for Improved Generalization in Supervised Learning Shahin Shahrampour, Ahmad Beirami, Vahid Tarokh

NeurIPS 2017 On Optimal Generalizability in Parametric Learning Ahmad Beirami, Meisam Razaviyayn, Shahin Shahrampour, Vahid Tarokh