CoT-ICL Lab: A Synthetic Framework for Studying Chain-of-Thought Learning from In-Context Demonstrations. - Kothapalli, Vignesh; Firooz, Hamed; Sanjabi, Maziar (2025)

360Brew: A Decoder-only Foundation Model for Personalized Ranking and Recommendation. - Firooz, Hamed; Sanjabi, Maziar; Englhardt, Adrian; Gupta, Aman; Levine, Ben; Olgiati, Dre; Polatkan, Gungor; Melnychuk, Iuliia; Ramgopal, Karthik; Talanine, Kirill; Srinivasan, Kutta; Simon, Luke; et al. (CoRR, 2025)

Efficient AI in Practice: Training and Deployment of Efficient LLMs for Industry Applications. - Behdin, Kayhan; Dai, Yun; Baarzi, Ata Fatahi; Gupta, Aman; Song, Qingquan; Tang, Shao; Sang, Hejian; Dexter, Gregory; Zhu, Sirou; Zhu, Siyu; Dharamsi, Tejas; Sanjabi, Maziar; et al. (CoRR, 2025)

CoT-ICL Lab: A Petri Dish for Studying Chain-of-Thought Learning from In-Context Demonstrations. - Kothapalli, Vignesh; Firooz, Hamed; Sanjabi, Maziar (CoRR, 2025)

Measuring Self-Supervised Representation Quality for Downstream Classification Using Discriminative Features. - Kalibhat, Neha Mukund; Narang, Kanika; Firooz, Hamed; Sanjabi, Maziar; Feizi, Soheil (2024)

Text-to-Sticker: Style Tailoring Latent Diffusion Models for Human Expression. - Sinha, Animesh; Sun, Bo; Kalia, Anmol; Casanova, Arantxa; Blanchard, Elliot; Yan, David; Zhang, Winnie; Nelli, Tony; Chen, Jiahui; Shah, Hardik; Yu, Licheng; Singh, Mitesh Kumar; et al. (2024)

Distilling Knowledge from Text-to-Image Generative Models Improves Visio-Linguistic Reasoning in CLIP. - Basu, Samyadeep; Hu, Shell Xu; Sanjabi, Maziar; Massiceti, Daniela; Feizi, Soheil (2024)

Differentially Private Representation Learning via Image Captioning. - Sander, Tom; Yu, Yaodong; Sanjabi, Maziar; Durmus, Alain Oliviero; Ma, Yi; Chaudhuri, Kamalika; Guo, Chuan (2024)

ViP: A Differentially Private Foundation Model for Computer Vision. - Yu, Yaodong; Sanjabi, Maziar; Ma, Yi; Chaudhuri, Kamalika; Guo, Chuan (2024)

RESPROMPT: Residual Connection Prompting Advances Multi-Step Reasoning in Large Language Models. - Jiang, Song; Shakeri, Zahra; Chan, Aaron; Sanjabi, Maziar; Firooz, Hamed; Xia, Yinglong; Akyildiz, Bugra; Sun, Yizhou; Li, Jinchao; Wang, Qifan; Celikyilmaz, Asli (2024)

Differentially Private Representation Learning via Image Captioning. - Sander, Tom; Yu, Yaodong; Sanjabi, Maziar; Durmus, Alain; Ma, Yi; Chaudhuri, Kamalika; Guo, Chuan (CoRR, 2024)

DP-RDM: Adapting Diffusion Models to Private Domains Without Fine-Tuning. - Lebensold, Jonathan; Sanjabi, Maziar; Astolfi, Pietro; Romero-Soriano, Adriana; Chaudhuri, Kamalika; Rabbat, Mike; Guo, Chuan (CoRR, 2024)

Lost-in-Distance: Impact of Contextual Proximity on LLM Performance in Graph Tasks. - Firooz, Hamed; Sanjabi, Maziar; Jiang, Wenlong; Zhai, Xiaoling (CoRR, 2024)

On Tilted Losses in Machine Learning: Theory and Applications. - Li, Tian; Beirami, Ahmad; Sanjabi, Maziar; Smith, Virginia (J. Mach. Learn. Res., 2023)

Text2Concept: Concept Activation Vectors Directly from Text. - Moayeri, Mazda; Rezaei, Keivan; Sanjabi, Maziar; Feizi, Soheil (2023)

Defending Against Patch-based Backdoor Attacks on Self-Supervised Learning. - Tejankar, Ajinkya; Sanjabi, Maziar; Wang, Qifan; Wang, Sinong; Firooz, Hamed; Pirsiavash, Hamed; Tan, Liang (2023)

COFFEE: Counterfactual Fairness for Personalized Text Generation in Explainable Recommendation. - Wang, Nan; Wang, Qifan; Wang, Yi-Chia; Sanjabi, Maziar; Liu, Jingzhou; Firooz, Hamed; Wang, Hongning; Nie, Shaoliang (2023)

Where to Begin? On the Impact of Pre-Training and Initialization in Federated Learning. - Nguyen, John; Wang, Jianyu; Malik, Kshitiz; Sanjabi, Maziar; Rabbat, Michael G. (2023)

Analyzing Privacy Leakage in Machine Learning via Multiple Hypothesis Testing: A Lesson From Fano. - Guo, Chuan; Sablayrolles, Alexandre; Sanjabi, Maziar (2023)

Identifying Interpretable Subspaces in Image Representations. - Kalibhat, Neha Mukund; Bhardwaj, Shweta; Bruss, C. Bayan; Firooz, Hamed; Sanjabi, Maziar; Feizi, Soheil (2023)

Text-To-Concept (and Back) via Cross-Model Alignment. - Moayeri, Mazda; Rezaei, Keivan; Sanjabi, Maziar; Feizi, Soheil (2023)

Text2Concept: Concept Activation Vectors Directly From Text. - Moayeri, Mazda; Rezaei, Keivan; Sanjabi, Maziar; Feizi, Soheil (2023)

Defending Against Patch-based Backdoor Attacks on Self-Supervised Learning. - Tejankar, Ajinkya; Sanjabi, Maziar; Wang, Qifan; Wang, Sinong; Firooz, Hamed; Pirsiavash, Hamed; Tan, Liang (CoRR, 2023)

Text-To-Concept (and Back) via Cross-Model Alignment. - Moayeri, Mazda; Rezaei, Keivan; Sanjabi, Maziar; Feizi, Soheil (CoRR, 2023)

ViP: A Differentially Private Foundation Model for Computer Vision. - Yu, Yaodong; Sanjabi, Maziar; Ma, Yi; Chaudhuri, Kamalika; Guo, Chuan (CoRR, 2023)

Augmenting CLIP with Improved Visio-Linguistic Reasoning. - Basu, Samyadeep; Sanjabi, Maziar; Massiceti, Daniela; Hu, Shell Xu; Feizi, Soheil (CoRR, 2023)

Identifying Interpretable Subspaces in Image Representations. - Kalibhat, Neha Mukund; Bhardwaj, Shweta; Bruss, C. Bayan; Firooz, Hamed; Sanjabi, Maziar; Feizi, Soheil (CoRR, 2023)

EditVal: Benchmarking Diffusion Based Text-Guided Image Editing Methods. - Basu, Samyadeep; Saberi, Mehrdad; Bhardwaj, Shweta; Chegini, Atoosa Malemir; Massiceti, Daniela; Sanjabi, Maziar; Hu, Shell Xu; Feizi, Soheil (CoRR, 2023)

Resprompt: Residual Connection Prompting Advances Multi-Step Reasoning in Large Language Models. - Jiang, Song; Shakeri, Zahra; Chan, Aaron; Sanjabi, Maziar; Firooz, Hamed; Xia, Yinglong; Akyildiz, Bugra; Sun, Yizhou; Li, Jinchao; Wang, Qifan; Celikyilmaz, Asli (CoRR, 2023)

Text-to-Sticker: Style Tailoring Latent Diffusion Models for Human Expression. - Sinha, Animesh; Sun, Bo; Kalia, Anmol; Casanova, Arantxa; Blanchard, Elliot; Yan, David; Zhang, Winnie; Nelli, Tony; Chen, Jiahui; Shah, Hardik; Yu, Licheng; Singh, Mitesh Kumar; et al. (CoRR, 2023)

FedShuffle: Recipes for Better Use of Local Work in Federated Learning. - Horváth, Samuel; Sanjabi, Maziar; Xiao, Lin; Richtárik, Peter; Rabbat, Michael G. (Trans. Mach. Learn. Res., 2022)

Detection, Disambiguation, Re-ranking: Autoregressive Entity Linking as a Multi-Task Problem. - Mrini, Khalil; Nie, Shaoliang; Gu, Jiatao; Wang, Sinong; Sanjabi, Maziar; Firooz, Hamed (2022)

ER-Test: Evaluating Explanation Regularization Methods for Language Models. - Joshi, Brihi; Chan, Aaron; Liu, Ziyi; Nie, Shaoliang; Sanjabi, Maziar; Firooz, Hamed; Ren, Xiang (2022)

UNIREX: A Unified Learning Framework for Language Model Rationale Extraction. - Chan, Aaron; Sanjabi, Maziar; Mathias, Lambert; Tan, Liang; Nie, Shaoliang; Peng, Xiaochang; Ren, Xiang; Firooz, Hamed (2022)

Federated Learning with Partial Model Personalization. - Pillutla, Krishna; Malik, Kshitiz; Mohamed, Abdelrahman; Rabbat, Michael G.; Sanjabi, Maziar; Xiao, Lin (2022)

BARACK: Partially Supervised Group Robustness With Guarantees. - Sohoni, Nimit Sharad; Sanjabi, Maziar; Ballas, Nicolas; Grover, Aditya; Nie, Shaoliang; Firooz, Hamed; Ré, Christopher (CoRR, 2022)

Understanding Failure Modes of Self-Supervised Learning. - Kalibhat, Neha Mukund; Narang, Kanika; Tan, Liang; Firooz, Hamed; Sanjabi, Maziar; Feizi, Soheil (CoRR, 2022)

Federated Learning with Partial Model Personalization. - Pillutla, Krishna; Malik, Kshitiz; Mohamed, Abdelrahman; Rabbat, Michael G.; Sanjabi, Maziar; Xiao, Lin (CoRR, 2022)

Detection, Disambiguation, Re-ranking: Autoregressive Entity Linking as a Multi-Task Problem. - Mrini, Khalil; Nie, Shaoliang; Gu, Jiatao; Wang, Sinong; Sanjabi, Maziar; Firooz, Hamed (CoRR, 2022)

FedShuffle: Recipes for Better Use of Local Work in Federated Learning. - Horváth, Samuel; Sanjabi, Maziar; Xiao, Lin; Richtárik, Peter; Rabbat, Michael G. (CoRR, 2022)

ER-TEST: Evaluating Explanation Regularization Methods for NLP Models. - Joshi, Brihi; Chan, Aaron; Liu, Ziyi; Nie, Shaoliang; Sanjabi, Maziar; Firooz, Hamed; Ren, Xiang (CoRR, 2022)

Where to Begin? Exploring the Impact of Pre-Training and Initialization in Federated Learning. - Nguyen, John; Malik, Kshitiz; Sanjabi, Maziar; Rabbat, Michael G. (CoRR, 2022)

FRAME: Evaluating Simulatability Metrics for Free-Text Rationales. - Chan, Aaron; Nie, Shaoliang; Tan, Liang; Peng, Xiaochang; Firooz, Hamed; Sanjabi, Maziar; Ren, Xiang (CoRR, 2022)

Where to Begin? On the Impact of Pre-Training and Initialization in Federated Learning. - Nguyen, John; Wang, Jianyu; Malik, Kshitiz; Sanjabi, Maziar; Rabbat, Michael G. (CoRR, 2022)

Analyzing Privacy Leakage in Machine Learning via Multiple Hypothesis Testing: A Lesson From Fano. - Guo, Chuan; Sablayrolles, Alexandre; Sanjabi, Maziar (CoRR, 2022)

COFFEE: Counterfactual Fairness for Personalized Text Generation in Explainable Recommendation. - Wang, Nan; Nie, Shaoliang; Wang, Qifan; Wang, Yi-Chia; Sanjabi, Maziar; Liu, Jingzhou; Firooz, Hamed; Wang, Hongning (CoRR, 2022)

Alternating Direction Method of Multipliers for Quantization. - Huang, Tianjian; Singhania, Prajwal; Sanjabi, Maziar; Mitra, Pabitra; Razaviyayn, Meisam (2021)

MSD: Saliency-aware Knowledge Distillation for Multimodal Understanding. - Jin, Woojeong; Sanjabi, Maziar; Nie, Shaoliang; Tan, Liang; Ren, Xiang; Firooz, Hamed (2021)

Tilted Empirical Risk Minimization. - Li, Tian; Beirami, Ahmad; Sanjabi, Maziar; Smith, Virginia (2021)

Modality-specific Distillation. - Jin, Woojeong; Sanjabi, Maziar; Nie, Shaoliang; Tan, Liang; Ren, Xiang; Firooz, Hamed (CoRR, 2021)

On Tilted Losses in Machine Learning: Theory and Applications. - Li, Tian; Beirami, Ahmad; Sanjabi, Maziar; Smith, Virginia (CoRR, 2021)

UniREx: A Unified Learning Framework for Language Model Rationale Extraction. - Chan, Aaron; Sanjabi, Maziar; Mathias, Lambert; Tan, Liang; Nie, Shaoliang; Peng, Xiaochang; Ren, Xiang; Firooz, Hamed (CoRR, 2021)

A Fistful of Words: Learning Transferable Visual Models from Bag-of-Words Supervision. - Tejankar, Ajinkya; Sanjabi, Maziar; Wu, Bichen; Xie, Saining; Khabsa, Madian; Pirsiavash, Hamed; Firooz, Hamed (CoRR, 2021)

Nonconvex Min-Max Optimization: Applications, Challenges, and Recent Theoretical Advances. - Razaviyayn, Meisam; Huang, Tianjian; Lu, Songtao; Nouiehed, Maher; Sanjabi, Maziar; Hong, Mingyi (IEEE Signal Process. Mag., 2020)

Building Placements In Urban Modeling Using Conditional Generative Latent Optimization. - Liang, Jingwen; Liu, Han; Zhao, Yiwei; Sanjabi, Maziar; Sardari, Mohsen; Chaput, Harold; Aghdaie, Navid; Zaman, Kazi A. (2020)

Fair Resource Allocation in Federated Learning. - Li, Tian; Sanjabi, Maziar; Beirami, Ahmad; Smith, Virginia (2020)

Federated Optimization in Heterogeneous Networks. - Li, Tian; Sahu, Anit Kumar; Zaheer, Manzil; Sanjabi, Maziar; Talwalkar, Ameet; Smith, Virginia (2020)

FedDANE: A Federated Newton-Type Method. - Li, Tian; Sahu, Anit Kumar; Zaheer, Manzil; Sanjabi, Maziar; Talwalkar, Ameet; Smith, Virginia (CoRR, 2020)

Non-convex Min-Max Optimization: Applications, Challenges, and Recent Theoretical Advances. - Razaviyayn, Meisam; Huang, Tianjian; Lu, Songtao; Nouiehed, Maher; Sanjabi, Maziar; Hong, Mingyi (CoRR, 2020)

Tilted Empirical Risk Minimization. - Li, Tian; Beirami, Ahmad; Sanjabi, Maziar; Smith, Virginia (CoRR, 2020)

Alternating Direction Method of Multipliers for Quantization. - Huang, Tianjian; Singhania, Prajwal; Sanjabi, Maziar; Mitra, Pabitra; Razaviyayn, Meisam (CoRR, 2020)

Multi-theme generative adversarial terrain amplification. - Zhao, Yiwei; Liu, Han; Borovikov, Igor; Beirami, Ahmad; Sanjabi, Maziar; Zaman, Kazi A. (ACM Trans. Graph., 2019)

FedDANE: A Federated Newton-Type Method. - Li, Tian; Sahu, Anit Kumar; Zaheer, Manzil; Sanjabi, Maziar; Talwalkar, Ameet; Smith, Virginia (2019)

Training Generative Networks Using Random Discriminators. - Barazandeh, Babak; Razaviyayn, Meisam; Sanjabi, Maziar (2019)

Solving a Class of Non-Convex Min-Max Games Using Iterative First Order Methods. - Nouiehed, Maher; Sanjabi, Maziar; Huang, Tianjian; Lee, Jason D.; Razaviyayn, Meisam (2019)

Solving a Class of Non-Convex Min-Max Games Using Iterative First Order Methods. - Nouiehed, Maher; Sanjabi, Maziar; Huang, Tianjian; Lee, Jason D.; Razaviyayn, Meisam (CoRR, 2019)

Training generative networks using random discriminators. - Barazandeh, Babak; Razaviyayn, Meisam; Sanjabi, Maziar (CoRR, 2019)

Fair Resource Allocation in Federated Learning. - Li, Tian; Sanjabi, Maziar; Smith, Virginia (CoRR, 2019)

When Does Non-Orthogonal Tensor Decomposition Have No Spurious Local Minima? - Sanjabi, Maziar; Baharlouei, Sina; Razaviyayn, Meisam; Lee, Jason D. (CoRR, 2019)

On the Convergence and Robustness of Training GANs with Regularized Optimal Transport. - Sanjabi, Maziar; Ba, Jimmy; Razaviyayn, Meisam; Lee, Jason D. (2018)

Solving Approximate Wasserstein GANs to Stationarity. - Sanjabi, Maziar; Ba, Jimmy; Razaviyayn, Meisam; Lee, Jason D. (CoRR, 2018)

Solving Non-Convex Non-Concave Min-Max Games Under Polyak-Łojasiewicz Condition. - Sanjabi, Maziar; Razaviyayn, Meisam; Lee, Jason D. (CoRR, 2018)

On the Convergence of Federated Optimization in Heterogeneous Networks. - Sahu, Anit Kumar; Li, Tian; Sanjabi, Maziar; Zaheer, Manzil; Talwalkar, Ameet; Smith, Virginia (CoRR, 2018)

Federated Multi-Task Learning. - Smith, Virginia; Chiang, Chao-Kai; Sanjabi, Maziar; Talwalkar, Ameet (2017)

Federated Multi-Task Learning. - Smith, Virginia; Chiang, Chao-Kai; Sanjabi, Maziar; Talwalkar, Ameet (CoRR, 2017)

A Stochastic Successive Minimization Method for Nonsmooth Nonconvex Optimization with Applications to Transceiver Design in Wireless Communication Networks. - Razaviyayn, Meisam; Sanjabi, Maziar; Luo, Zhi-Quan (Math. Program., 2016)

Accelerated Alternating Direction Method of Multipliers. - Kadkhodaie, Mojtaba; Christakopoulou, Konstantina; Sanjabi, Maziar; Banerjee, Arindam (2015)

Cross-Layer Provision of Future Cellular Networks: A WMMSE-based approach. - Baligh, Hadi; Hong, Mingyi; Liao, Wei-Cheng; Luo, Zhi-Quan; Razaviyayn, Meisam; Sanjabi, Maziar; Sun, Ruoyu (IEEE Signal Process. Mag., 2014)

Optimal Joint Base Station Assignment and Beamforming for Heterogeneous Networks. - Sanjabi, Maziar; Razaviyayn, Meisam; Luo, Zhi-Quan (IEEE Trans. Signal Process., 2014)

Joint base station clustering and beamformer design for partial coordinated transmission using statistical channel state information. - Sanjabi, Maziar; Hong, Mingyi; Razaviyayn, Meisam; Luo, Zhi-Quan (2014)

Cross Layer Provision of Future Cellular Networks. - Baligh, Hadi; Hong, Mingyi; Liao, Wei-Cheng; Luo, Zhi-Quan; Razaviyayn, Meisam; Sanjabi, Maziar; Sun, Ruoyu (CoRR, 2014)

A stochastic weighted MMSE approach to sum rate maximization for a MIMO interference channel. - Razaviyayn, Meisam; Boroujeni, Maziar Sanjabi; Luo, Zhi-Quan (2013)

A Stochastic Successive Minimization Method for Nonsmooth Nonconvex Optimization with Applications to Transceiver Design in Wireless Communication Networks. - Razaviyayn, Meisam; Sanjabi, Maziar; Luo, Zhi-Quan (CoRR, 2013)

Robust SINR-Constrained MISO Downlink Beamforming: When is Semidefinite Programming Relaxation Tight? - Song, Enbin; Shi, Qingjiang; Sanjabi, Maziar; Sun, Ruoyu; Luo, Zhi-Quan (EURASIP J. Wirel. Commun. Netw., 2012)

Linear Transceiver Design for Interference Alignment: Complexity and Computation. - Razaviyayn, Meisam; Sanjabi, Maziar; Luo, Zhi-Quan (IEEE Trans. Inf. Theory, 2012)

Optimal joint base station assignment and downlink beamforming for heterogeneous networks. - Sanjabi, Maziar; Razaviyayn, Meisam; Luo, Zhi-Quan (2012)

Robust SINR-constrained MISO downlink beamforming: When is semidefinite programming relaxation tight? - Song, Enbin; Shi, Qingjiang; Sanjabi, Maziar; Sun, Ruoyu; Luo, Zhi-Quan (2011)

Linear Transceiver Design for Interference Alignment: Complexity and Computation - Razaviyayn, Meisam; Sanjabi, Maziar; Luo, Zhi-Quan (CoRR, 2010)