Soltanolkotabi, Mahdi

45 publications

NeurIPS 2025 Emergence and Evolution of Interpretable Concepts in Diffusion Models Berk Tinaz, Zalan Fabian, Mahdi Soltanolkotabi

CVPRW 2025 HARMONY: Hidden Activation Representations and Model Output-Aware Uncertainty Estimation for Vision-Language Models Erum Mushtaq, Zalan Fabian, Yavuz Faruk Bakman, Anil Ramakrishna, Mahdi Soltanolkotabi, Salman Avestimehr

NeurIPS 2025 Hyperphantasia: A Benchmark for Evaluating the Mental Visualization Capabilities of Multimodal LLMs Mohammad Shahab Sepehri, Berk Tinaz, Zalan Fabian, Mahdi Soltanolkotabi

ICLR 2025 MediConfusion: Can You Trust Your AI Radiologist? Probing the Reliability of Multimodal Medical Foundation Models Mohammad Shahab Sepehri, Zalan Fabian, Maryam Soltanolkotabi, Mahdi Soltanolkotabi

ICML 2025 Test-Time Training Provably Improves Transformers as In-Context Learners Halil Alperen Gozeten, Muhammed Emrullah Ildiz, Xuechen Zhang, Mahdi Soltanolkotabi, Marco Mondelli, Samet Oymak

NeurIPS 2025 The Rich and the Simple: On the Implicit Bias of Adam and SGD Bhavya Vasudeva, Jung Whan Lee, Vatsal Sharan, Mahdi Soltanolkotabi

TMLR 2025 Theoretical Insights into Overparameterized Models in Multi-Task and Replay-Based Continual Learning Mohammadamin Banayeeanzade, Mahdi Soltanolkotabi, Mohammad Rostami

ICML 2024 Adapt and Diffuse: Sample-Adaptive Reconstruction via Latent Diffusion Models Zalan Fabian, Berk Tinaz, Mahdi Soltanolkotabi

ICML 2024 DiracDiffusion: Denoising and Incremental Reconstruction with Assured Data-Consistency Zalan Fabian, Berk Tinaz, Mahdi Soltanolkotabi

NeurIPSW 2024 Implicit Bias of Adam Versus Gradient Descent in One-Hidden-Layer Neural Networks Bhavya Vasudeva, Vatsal Sharan, Mahdi Soltanolkotabi

JMLR 2024 Learning from Many Trajectories Stephen Tu, Roy Frostig, Mahdi Soltanolkotabi

NeurIPSW 2024 MediConfusion: Can You Trust Your AI Radiologist? Probing the Reliability of Multimodal Medical Foundation Models Mohammad Shahab Sepehri, Zalan Fabian, Maryam Soltanolkotabi, Mahdi Soltanolkotabi

NeurIPSW 2024 MediConfusion: Can You Trust Your AI Radiologist? Probing the Reliability of Multimodal Medical Foundation Models Mohammad Shahab Sepehri, Zalan Fabian, Maryam Soltanolkotabi, Mahdi Soltanolkotabi

ICML 2024 Provable Multi-Task Representation Learning by Two-Layer ReLU Neural Networks Liam Collins, Hamed Hassani, Mahdi Soltanolkotabi, Aryan Mokhtari, Sanjay Shakkottai

NeurIPS 2023 A Data-Free Approach to Mitigate Catastrophic Forgetting in Federated Class Incremental Learning for Vision Tasks Sara Babakniya, Zalan Fabian, Chaoyang He, Mahdi Soltanolkotabi, Salman Avestimehr

NeurIPSW 2023 Adapt and Diffuse: Sample-Adaptive Reconstruction via Latent Diffusion Models Zalan Fabian, Berk Tinaz, Mahdi Soltanolkotabi

CVPR 2023 CUDA: Convolution-Based Unlearnable Datasets Vinu Sankar Sadasivan, Mahdi Soltanolkotabi, Soheil Feizi

ICMLW 2023 Don’t Memorize; Mimic the past: Federated Class Incremental Learning Without Episodic Memory Sara Babakniya, Zalan Fabian, Chaoyang He, Mahdi Soltanolkotabi, Salman Avestimehr

COLT 2023 Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensing Mahdi Soltanolkotabi, Dominik Stöger, Changzhi Xie

NeurIPS 2023 Learning Provably Robust Estimators for Inverse Problems via Jittering Anselm Krainovic, Mahdi Soltanolkotabi, Reinhard Heckel

ICML 2023 On the Role of Attention in Prompt-Tuning Samet Oymak, Ankit Singh Rawat, Mahdi Soltanolkotabi, Christos Thrampoulidis

ICLRW 2023 On the Role of Attention in Prompt-Tuning Samet Oymak, Ankit Singh Rawat, Mahdi Soltanolkotabi, Christos Thrampoulidis

TMLR 2023 mL-BFGS: A Momentum-Based L-BFGS for Distributed Large-Scale Neural Network Optimization Yue Niu, Zalan Fabian, Sunwoo Lee, Mahdi Soltanolkotabi, Salman Avestimehr

NeurIPS 2022 HUMUS-Net: Hybrid Unrolled Multi-Scale Network Architecture for Accelerated MRI Reconstruction Zalan Fabian, Berk Tinaz, Mahdi Soltanolkotabi

COLT 2022 Neural Networks Can Learn Representations with Gradient Descent Alexandru Damian, Jason Lee, Mahdi Soltanolkotabi

NeurIPS 2022 Outlier-Robust Sparse Estimation via Non-Convex Optimization Yu Cheng, Ilias Diakonikolas, Rong Ge, Shivam Gupta, Daniel Kane, Mahdi Soltanolkotabi

ICML 2021 Data Augmentation for Deep Learning Based Accelerated MRI Reconstruction with Limited Data Zalan Fabian, Reinhard Heckel, Mahdi Soltanolkotabi

ICML 2021 Generalization Guarantees for Neural Architecture Search with Train-Validation Split Samet Oymak, Mingchen Li, Mahdi Soltanolkotabi

ICML 2021 PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-Scale Models Chaoyang He, Shen Li, Mahdi Soltanolkotabi, Salman Avestimehr

NeurIPS 2021 Small Random Initialization Is Akin to Spectral Learning: Optimization and Generalization Guarantees for Overparameterized Low-Rank Matrix Reconstruction Dominik Stöger, Mahdi Soltanolkotabi

ICLR 2021 Understanding Over-Parameterization in Generative Adversarial Networks Yogesh Balaji, Mohammadmahdi Sajedi, Neha Mukund Kalibhat, Mucong Ding, Dominik Stöger, Mahdi Soltanolkotabi, Soheil Feizi

COLT 2020 Approximation Schemes for ReLU Regression Ilias Diakonikolas, Surbhi Goel, Sushrut Karmalkar, Adam R. Klivans, Mahdi Soltanolkotabi

ICML 2020 Compressive Sensing with Un-Trained Neural Networks: Gradient Descent Finds a Smooth Approximation Reinhard Heckel, Mahdi Soltanolkotabi

ICLR 2020 Denoising and Regularization via Exploiting the Structural Bias of Convolutional Generators Reinhard Heckel, Mahdi Soltanolkotabi

AISTATS 2020 Gradient Descent with Early Stopping Is Provably Robust to Label Noise for Overparameterized Neural Networks Mingchen Li, Mahdi Soltanolkotabi, Samet Oymak

ICML 2020 High-Dimensional Robust Mean Estimation via Gradient Descent Yu Cheng, Ilias Diakonikolas, Rong Ge, Mahdi Soltanolkotabi

L4DC 2020 Learning the Model-Free Linear Quadratic Regulator via Random Search Hesameddin Mohammadi, Mihailo R. Jovanovic’, Mahdi Soltanolkotabi

NeurIPS 2020 Minimax Lower Bounds for Transfer Learning with Linear and One-Hidden Layer Neural Networks Mohammadreza Mousavi Kalan, Zalan Fabian, Salman Avestimehr, Mahdi Soltanolkotabi

COLT 2020 Precise Tradeoffs in Adversarial Training for Linear Regression Adel Javanmard, Mahdi Soltanolkotabi, Hamed Hassani

NeurIPS 2020 Theoretical Insights into Multiclass Classification: A High-Dimensional Asymptotic View Christos Thrampoulidis, Samet Oymak, Mahdi Soltanolkotabi

AISTATS 2019 Lagrange Coded Computing: Optimal Design for Resiliency, Security, and Privacy Qian Yu, Songze Li, Netanel Raviv, Seyed Mohammadreza Mousavi Kalan, Mahdi Soltanolkotabi, Salman A. Avestimehr

ICML 2019 Overparameterized Nonlinear Learning: Gradient Descent Takes the Shortest Path? Samet Oymak, Mahdi Soltanolkotabi

NeurIPS 2017 Gradient Methods for Submodular Maximization Hamed Hassani, Mahdi Soltanolkotabi, Amin Karbasi

NeurIPS 2017 Learning ReLUs via Gradient Descent Mahdi Soltanolkotabi

ICML 2016 Low-Rank Solutions of Linear Matrix Equations via Procrustes Flow Stephen Tu, Ross Boczar, Max Simchowitz, Mahdi Soltanolkotabi, Ben Recht