← Search

Mohammad Kachuee

10 accepted papers

2026

TruthRL: Incentivizing Truthful LLMs via Reinforcement Learning

ICML 2026poster

While large language models (LLMs) have demonstrated strong performance on factoid question answering, they are still prone to hallucination and untruthful responses, particularly when tasks demand information outside their parametric knowledge. Indeed, truthfulness requires more than accuracy---mod…

Cited by 0SourceScholar
2025

Improving Tool Retrieval by Leveraging Large Language Models for Query Generation

COLING 2025industry

Using tools by Large Language Models (LLMs) is a promising avenue to extend their reach beyond language or conversational settings. The number of tools can scale to thousands as they enable accessing sensory information, fetching updated factual knowledge, or taking actions in the real world. In suc…

Cited by 1SourcePDFScholar
2024

LLM-based Frameworks for API Argument Filling in Task-Oriented Conversational Systems

NAACL 2024industry

Task-orientated conversational agents interact with users and assist them via leveraging external APIs. A typical task-oriented conversational system can be broken down into three phases: external API selection, argument filling, and response generation. The focus of our work is the task of argument…

Cited by 4SourcePDFScholar
2024

Planning and Editing What You Retrieve for Enhanced Tool Learning

NAACL 2024findings

Recent advancements in integrating external tools with Large Language Models (LLMs) have opened new frontiers, with applications in mathematical reasoning, code generators, and smart assistants. However, existing methods, relying on simple one-time retrieval strategies, fall short on effectively and…

2024

Skin-in-the-Game: Decision Making via Multi-Stakeholder Alignment in LLMs

ACL 2024long

Large Language Models (LLMs) have shown remarkable capabilities in tasks such as summarization, arithmetic reasoning, and question answering. However, they encounter significant challenges in the domain of moral reasoning and ethical decision-making, especially in complex scenarios with multiple sta…

Cited by 6SourcePDFScholar
2023

Constrained Policy Optimization for Controlled Self-Learning in Conversational AI Systems

ACL 2023industry

Recently, self-learning methods based on user satisfaction metrics and contextual bandits have shown promising results to enable consistent improvements in conversational AI systems. However, directly targeting such metrics by off-policy bandit learning objectives often increases the risk of making…

Cited by 2SourcePDFScholar
2023

Scalable and Safe Remediation of Defective Actions in Self-Learning Conversational Systems

ACL 2023industry

Off-Policy reinforcement learning has been the driving force for the state-of-the-art conversational AIs leading to more natural human-agent interactions and improving the user satisfaction for goal-oriented agents. However, in large-scale commercial settings, it is often challenging to balance betw…

Cited by 0SourcePDFScholar
2022

Scalable and Robust Self-Learning for Skill Routing in Large-Scale Conversational AI Systems

NAACL 2022industry

Skill routing is an important component in large-scale conversational systems. In contrast to traditional rule-based skill routing, state-of-the-art systems use a model-based approach to enable natural conversations. To provide supervision signal required to train such models, ideas such as human an…

Cited by 3SourcePDFScholar
2021

Self-Supervised Contrastive Learning for Efficient User Satisfaction Prediction in Conversational Agents

NAACL 2021long

Turn-level user satisfaction is one of the most important performance metrics for conversational agents. It can be used to monitor the agent’s performance and provide insights about defective user experiences. While end-to-end deep learning has shown promising results, having access to a large numbe…

Cited by 34SourcePDFScholar
2019

Opportunistic Learning: Budgeted Cost-Sensitive Learning from Data Streams

ICLR 2019poster

In many real-world learning scenarios, features are only acquirable at a cost constrained under a budget. In this paper, we propose a novel approach for cost-sensitive feature acquisition at the prediction-time. The suggested method acquires features incrementally based on a context-aware feature-va…