Papers by Mohsen Bayati
Text-to-Distribution Prediction with Quantile Tokens and Neighbor Context (2026.acl-long)
Copied to clipboard
Yilun Zhu, Yuan Zhuang, Nikhita Vedula, Dushyanta Dhyani, Shaoyuan Xu, Mohsen Bayati, Bryan Wang, Shervin Malmasi
| Challenge: | Existing methods for text regression lack local grounding and rely on shared representations. |
| Approach: | They propose a distributional regression model with quantile tokens that insert dedicated quantiles into the input sequence. |
| Outcome: | The proposed method outperforms baseline models on the inside Airbnb and StackSample datasets. |
Quantile Regression with Large Language Models for Price Prediction (2025.findings-acl)
Copied to clipboard
| Challenge: | Existing approaches to structured prediction tasks focus on point estimates and lack systematic comparison across different methods. |
| Approach: | They propose a novel quantile regression approach that enables LLMs to produce full predictive distributions, improving upon traditional point estimates. |
| Outcome: | The proposed model outperforms encoder architectures, embedding-based methods, and few-shot learning methods in prediction accuracy and distributional calibration. |