Publications

My current research focuses on LLM reasoning, self-improvement, and post-training. Over the years, I have also worked on generative recommender systems, geometric deep learning, bio-inspired optimization, and autonomous agents. Below is a selection of my publications organized by research area.

LLM Reasoning & Alignment

Paper teaser

Credit Assignment with Resets in Language Model Reasoning
Ankur Samanta, Akshayaa Magesh, Ayush Jain, Youliang Yu, Daniel Jiang, Kavosh Asadi, Kaveh Hassani, Paul Sajda, Jalaj Bhandari, Yonathan Efroni
Arxiv Preprint, 2026

Paper teaser

Imbalanced Gradients in RL Post-Training of Multi-Task LLMs
Runzhe Wu, Ankur Samanta, Ayush Jain, Scott Fujimoto, Jeongyeol Kwon, Ben Kretzu, Youliang Yu, Kaveh Hassani, Boris Vidolov, Yonathan Efroni
Findings of the Association for Computational Linguistics: EACL, 2026

Paper teaser

How to Make LLMs Strong Node Classifiers?
Zhe Xu, Kaveh Hassani, Si Zhang, Hanqing Zeng, Michihiro Yasunaga, Limei Wang, Dongqi Fu, Ning Yao, Bo Long, Hanghang Tong
Findings of the Association for Computational Linguistics: EACL, 2026

Paper teaser

Structure Enables Effective Self-Localization of Errors in LLMs
Ankur Samanta, Akshayaa Magesh, Ayush Jain, Kavosh Asadi, Youliang Yu, Daniel Jiang, Boris Vidolov, Kaveh Hassani, Paul Sajda, Jalaj Bhandari, Yonathan Efroni
International Conference on Machine Learning (ICML), 2026

Paper teaser

Self-Improvement of Language Models by Post-Training on Multi-Agent Debate
Ankur Samanta, Akshayaa Magesh, Runzhe Wu, Ayush Jain, Youliang Yu, Daniel Jiang, Boris Vidolov, Paul Sajda, Yonathan Efroni, Kaveh Hassani
Arxiv Preprint, 2025

Generative Recommender Systems

Paper teaser

Multimodal Generative Recommendation for Fusing Semantic and Collaborative Signals
Moritz Vandenhirtz, Kaveh Hassani, Shervin Ghasemlou, Shuai Shao, Hamid Eghbalzadeh, Fuchun Peng, Jun Liu, Michael Louis Iuzzolino
Arxiv Preprint, 2026

Paper teaser

Generating Long Semantic IDs in Parallel for Recommendation
Yupeng Hou, Jiacheng Li, Ashley Shin, Jinsung Jeon, Abhishek Santhanam, Wei Shao, Kaveh Hassani, Ning Yao, Julian McAuley
ACM Conference on Knowledge Discovery and Data Mining (KDD), 2025

Paper teaser

Billion-Scale Graph Deep Learning Framework for Ads Recommendation
Si Zhang, Weilin Cong, Dongqi Fu, Andrey Malevich, Hao Wu, Baichuan Yuan, Xin Zhou, Kaveh Hassani, Zhigang Hua, Austin Derrow-Pinion, Yan Xie, Xuewei Wang, Yinglong Xia, Ning Yao, Vena Li, Sem Park, Bo Long
ACM International Conference on Information and Knowledge Management (CIKM), 2025

Paper teaser

Preference Discerning with LLM-Enhanced Generative Retrieval
Fabian Paischer, Liu Yang, Linfeng Liu, Shuai Shao, Kaveh Hassani, Jiacheng Li, Ricky Chen, Zhang Gabriel Li, Xialo Gao, Wei Shao, Xue Feng, Nima Noorshams, Sem Park, Bo Long, Hamid Eghbalzadeh
Transactions on Machine Learning Research (TMLR), 2024

Paper teaser

Unifying Generative and Dense Retrieval for Sequential Recommendation
Liu Yang, Fabian Paischer, Kaveh Hassani, Jiacheng Li, Shuai Shao, Zhang Gabriel Li, Yun He, Xue Feng, Nima Noorshams, Sem Park, Bo Long, Robert D Nowak, Xiaoli Gao, Hamid Eghbalzadeh
Transactions on Machine Learning Research (TMLR), 2024

Paper teaser

Rankitect: Ranking Architecture Search Battling World-class Engineers at Meta Scale
Wei Wen, Kuang-Hung Liu, Igor Fedorov, Xin Zhang, Hang Yin, Weiwei Chu, Kaveh Hassani, Mengying Sun, Jiang Liu, Xu Wang, Lin Jiang, Yuxin Chen, Buyun Zhang, Xi Liu, Dehua Cheng, Zhengxing Chen, Guang Zhao, Fangqiu Han, Jiyan Yang, Yuchen Hao, Liang Xiong, Wen-Yen Chen
Proceedings of the ACM Web Conference (WWW), 2024

Geometric Deep Learning

Paper teaser

Higher-Order Structure Boosts Link Prediction on Temporal Graphs
Jingzhe Liu, Zhigang Hua, Yan Xie, Bingheng Li, Harry Shomer, Yu Song, Kaveh Hassani, Jiliang Tang
Arxiv Preprint, 2025

Paper teaser

Learning Graph Quantized Tokenizers
Limei Wang, Kaveh Hassani, Si Zhang, Dongqi Fu, Baichuan Yuan, Weilin Cong, Zhigang Hua, Hao Wu, Ning Yao, Bo Long
International Conference on Learning Representations (ICLR), 2025

Paper teaser

Staleness-Based Subgraph Sampling for Training GNNs on Large-Scale Graphs
Limei Wang, Si Zhang, Hanqing Zeng, Hao Wu, Zhigang Hua, Kaveh Hassani, Andrey Malevich, Bo Long, Shuiwang Ji
IEEE International Conference on Big Data (BigData), 2025

Paper teaser

Evaluating Graph Generative Models with Contrastively Learned Features
Hamed Shirzad, Kaveh Hassani, Danica J Sutherland
Neural Information Processing Systems (NeurIPS), 2022

Paper teaser

Material Prediction for Design Automation Using Graph Representation Learning
Shijie Bian, Daniele Grandi, Kaveh Hassani, Elliot Sadler, Bodia Borijin, Axel Fernandes, Andrew Wang, Thomas Lu, Richard Otis, Nhut Ho, Bingbing Li
Design Automation Conference, 2022

Paper teaser

Cross-Domain Few-Shot Graph Classification
Kaveh Hassani
AAAI Conference on Artificial Intelligence, 2022

Paper teaser

Classifying Component Function in Product Assemblies With Graph Neural Networks
Vincenzo Ferrero, Bryony DuPont, Kaveh Hassani, Daniele Grandi
Journal of Mechanical Design, 2022

Paper teaser

Learning Graph Augmentations to Learn Graph Representations
Kaveh Hassani, Amir Hosein Khasahmadi
Arxiv Preprint, 2022

Paper teaser

Contrastive Multi-View Representation Learning on Graphs
Kaveh Hassani, Amir Hosein Khasahmadi
International Conference on Machine Learning (ICML), 2020

Paper teaser

PointMask: Towards Interpretable and Bias-Resilient Point Cloud Processing
Saeid Asgari Taghanaki, Kaveh Hassani, Pradeep Kumar Jayaraman, Amir Hosein Khasahmadi, Tonya Custis
International Conference on Machine Learning (ICML) Workshop on Human Interpretability in Machine Learning, 2020

Paper teaser

Memory-Based Graph Networks
Amir Hosein Khasahmadi, Kaveh Hassani, Parsa Moradi, Leo Lee, Quaid Morris
International Conference on Learning Representations (ICLR), 2020

Paper teaser

Relational Graph Representation Learning for Open-Domain Question Answering
Salvatore Vivona, Kaveh Hassani
Neural Information Processing Systems (NeurIPS) Workshop on Graph Representation Learning, 2019

Paper teaser

Unsupervised Multi-Task Feature Learning on Point Clouds
Kaveh Hassani, Mike Haley
International Conference on Computer Vision (ICCV), 2019

Bio-Inspired Optimization & Control Theory

Paper teaser

Simulating Collective Intelligence of Bio-Inspired Competing Agents
Aliakbar Asgari, Kaveh Hassani, Won-Sook Lee
Expert Systems with Applications, 2016

Paper teaser
Paper teaser

A Case Study on Collective Intelligence Based on Energy Flow
Kaveh Hassani, Aliakbar Asgari, Won-Sook Lee
IEEE International Conference on Evolving and Adaptive Intelligent Systems (EAIS), 2015

Paper teaser

An Incremental Framework for Classification of EEG Signals Using Quantum Particle Swarm Optimization
Kaveh Hassani, Won-Sook Lee
IEEE International Conference on Computational Intelligence and Virtual Environments, 2014

Paper teaser

Optimal Tuning of Linear Quadratic Regulators using Quantum Particle Swarm Optimization
Kaveh Hassani, Won-Sook Lee
International Conference of Control, Dynamic Systems, and Robotics, 2014

Paper teaser

An Incremental Parallel Particle Swarm Approach for Classification Rule Discovery from Dynamic Data
Kaveh Hassani, Won-Sook Lee
International Conference on Machine Learning and Applications (ICMLA), 2013

Autonomous Agents & Virtual Environments

Paper teaser

Disambiguating Spatial Prepositions Using Deep Convolutional Networks
Kaveh Hassani, Won-Sook Lee
AAAI Conference on Artificial Intelligence, 2017

Paper teaser
Paper teaser

Learning Physical Properties of Objects Using Gaussian Mixture Models
Kaveh Hassani, Won-Sook Lee
Canadian Conference on Artificial Intelligence, 2017

Paper teaser

Visualizing Natural Language Descriptions: A Survey
Kaveh Hassani, Won-Sook Lee
ACM Computing Surveys, 2016

Paper teaser

A Universal Architecture for Migrating Cognitive Agents: A Case Study on Automatic Animation Generation
Kaveh Hassani, Won-Sook Lee
Integrating Cognitive Architectures into Virtual Character Design (Book Chapter), 2016

Paper teaser

Adaptive Animation Generation Using Web Content Mining
Kaveh Hassani, Won-Sook Lee
IEEE International Conference on Evolving and Adaptive Intelligent Systems (EAIS), 2015

Paper teaser

On Designing Migrating Agents: From Autonomous Virtual Agents to Intelligent Robotic Systems
Kaveh Hassani, Won-Sook Lee
SIGGRAPH Asia, Autonomous Virtual Humans and Social Robot for Telepresence Workshop, 2014

Paper teaser
Paper teaser

Design and Implementation of an Intelligent Virtual Environment for Improving Speaking and Listening Skills
Kaveh Hassani, Ali Nahvi, Ali Ahmadi
Interactive Learning Environments, 2013

Paper teaser

Architectural Design and Implementation of Intelligent Embodied Conversational Agents Using Fuzzy Knowledge Base
Kaveh Hassani, Ali Nahvi, Ali Ahmadi
Journal of Intelligent & Fuzzy Systems, 2013

Paper teaser

A Software-in-the-Loop Simulation of an Intelligent MicroSatellite within a Virtual Environment
Kaveh Hassani, Won-Sook Lee
IEEE International Conference on Computational Intelligence and Virtual Environments, 2013