Bateni, MohammadHossein

25 publications

ICML 2025 Bipartite Ranking from Multiple Labels: On Loss Versus Label Aggregation Michal Lukasik, Lin Chen, Harikrishna Narasimhan, Aditya Krishna Menon, Wittawat Jitkrittum, Felix X. Yu, Sashank J. Reddi, Gang Fu, Mohammadhossein Bateni, Sanjiv Kumar

ICML 2025 DeepCrossAttention: Supercharging Transformer Residual Connections Mike Heddes, Adel Javanmard, Kyriakos Axiotis, Gang Fu, Mohammadhossein Bateni, Vahab Mirrokni

NeurIPS 2025 Replicable Online Pricing Kiarash Banihashem, Mohammadhossein Bateni, Hossein Esfandiari, Samira Goudarzi, MohammadTaghi Hajiaghayi

ICML 2025 Synthetic Text Generation for Training Large Language Models via Gradient Matching Dang Nguyen, Zeman Li, Mohammadhossein Bateni, Vahab Mirrokni, Meisam Razaviyayn, Baharan Mirzasoleiman

AISTATS 2024 A Scalable Algorithm for Individually Fair K-Means Clustering MohammadHossein Bateni, Vincent Cohen-Addad, Alessandro Epasto, Silvio Lattanzi

NeurIPS 2024 Efficient Centroid-Linkage Clustering MohammadHossein Bateni, Laxman Dhulipala, Willem Fletcher, Kishen N. Gowda, D Ellis Hershkowitz, Rajesh Jayaram, Jakub Łącki

COLT 2024 Metric Clustering and MST with Strong and Weak Distance Oracles MohammadHossein Bateni, Prathamesh Dharangutte, Rajesh Jayaram, Chen Wang

NeurIPS 2024 SequentialAttention++ for Block Sparsification: Differentiable Pruning Meets Combinatorial Optimization Taisuke Yasuda, Kyriakos Axiotis, Gang Fu, MohammadHossein Bateni, Vahab Mirrokni

ICCV 2023 Agile Modeling: From Concept to Classifier in Minutes Otilia Stretcu, Edward Vendrow, Kenji Hata, Krishnamurthy Viswanathan, Vittorio Ferrari, Sasan Tavakkol, Wenlei Zhou, Aditya Avinash, Emming Luo, Neil Gordon Alldrin, MohammadHossein Bateni, Gabriel Berger, Andrew Bunner, Chun-Ta Lu, Javier Rey, Giulia DeSalvo, Ranjay Krishna, Ariel Fuxman‎

NeurIPSW 2023 Agile Modeling: From Concept to Classifier in Minutes Otilia Stretcu, Edward Vendrow, Kenji Hata, Krishnamurthy Viswanathan, Vittorio Ferrari, Sasan Tavakkol, Wenlei Zhou, Aditya Avinash, Enming Luo, Neil Gordon Alldrin, Mohammadhossein Bateni, Gabriel Berger, Andrew Bunner, Chun-Ta Lu, Javier A Rey, Giulia DeSalvo, Ranjay Krishna, Ariel Fuxman

ICMLW 2023 Preference Elicitation for Music Recommendations Ofer Meshi, Jon Feldman, Li Yang, Ben Scheetz, Yanli Cai, Mohammadhossein Bateni, Corbyn Salisbury, Vikram Aggarwal, Craig Boutilier

ICLR 2023 Sequential Attention for Feature Selection Taisuke Yasuda, Mohammadhossein Bateni, Lin Chen, Matthew Fahrbach, Gang Fu, Vahab Mirrokni

ICMLW 2023 Sequential Attention for Feature Selection Taisuke Yasuda, Mohammadhossein Bateni, Lin Chen, Matthew Fahrbach, Gang Fu, Vahab Mirrokni

UAI 2023 SubMix: Learning to Mix Graph Sampling Heuristics Sami Abu-El-Haija, Joshua V. Dillon, Bahare Fatemi, Kyriakos Axiotis, Neslihan Bulut, Johannes Gasteiger, Bryan Perozzi, Mohammadhossein Bateni

TMLR 2023 Tackling Provably Hard Representative Selection via Graph Neural Networks Mehran Kazemi, Anton Tsitsulin, Hossein Esfandiari, Mohammadhossein Bateni, Deepak Ramachandran, Bryan Perozzi, Vahab Mirrokni

ICMLW 2023 Tackling Provably Hard Representative Selection viaGraph Neural Networks Mehran Kazemi, Anton Tsitsulin, Hossein Esfandiari, Mohammadhossein Bateni, Deepak Ramachandran, Bryan Perozzi, Vahab Mirrokni

ICMLW 2023 Tensor Proxies for Efficient Feature Cross Search Taisuke Yasuda, Mohammadhossein Bateni, Lin Chen, Matthew Fahrbach, Gang Fu

NeurIPSW 2022 Scalable and Improved Algorithms for Individually Fair Clustering Mohammadhossein Bateni, Vincent Cohen-Addad, Alessandro Epasto, Silvio Lattanzi

AAAI 2021 Extreme K-Center Clustering MohammadHossein Bateni, Hossein Esfandiari, Manuela Fischer, Vahab S. Mirrokni

NeurIPS 2021 Streaming Belief Propagation for Community Detection Yuchen Wu, Jakab Tardos, Mohammadhossein Bateni, André Linhares, Filipe Miguel Goncalves de Almeida, Andrea Montanari, Ashkan Norouzi-Fard

ICML 2019 Categorical Feature Compression via Submodular Optimization Mohammadhossein Bateni, Lin Chen, Hossein Esfandiari, Thomas Fu, Vahab Mirrokni, Afshin Rostamizadeh

ICML 2019 Distributed Weighted Matching via Randomized Composable Coresets Sepehr Assadi, Mohammadhossein Bateni, Vahab Mirrokni

AAAI 2017 A Study of Compact Reserve Pricing Languages MohammadHossein Bateni, Hossein Esfandiari, Vahab S. Mirrokni, Saeed Seddighin

NeurIPS 2017 Affinity Clustering: Hierarchical Clustering at Scale Mohammadhossein Bateni, Soheil Behnezhad, Mahsa Derakhshan, MohammadTaghi Hajiaghayi, Raimondas Kiveris, Silvio Lattanzi, Vahab Mirrokni

NeurIPS 2014 Distributed Balanced Clustering via Mapping Coresets Mohammadhossein Bateni, Aditya Bhaskara, Silvio Lattanzi, Vahab Mirrokni