COLLT: A Multi-Task Optimization Framework for Clarification-Oriented Tool Learning in Legal Large Language Models

Tool-augmented Large Language Models (LLMs) have demonstrated remarkable capabilities in language understanding and generation across various domains, with notable progress in legal applications. However, existing legal LLMs still face two major challenges in legal question answering: (1) a persistent occurrence of hallucinations in generated legal advice and (2) limited effectiveness in handling ambiguous legal queries. To address both issues, this article introduces COLLT (Clarification-Oriented Legal Large language models enhanced by Tool learning), a legal LLM framework designed around a tool learning mechanism oriented to clarification. We model the clarification, tooling, and response workflow as a budgeted sequential decision. All components, including action selection, tool invocation, and response generation, are jointly optimized through multi-task instruction tuning. The COLLT framework first determines whether a user query contains ambiguity. If ambiguity is detected, the model actively guides the user to clarify their intent; if the query is clear, the model automatically invokes appropriate legal tools to improve the accuracy and reliability of its response. To support this mechanism, we propose an end-to-end training strategy enabling the model to learn how to invoke tools adaptively across various scenarios. Specifically, we construct an instruction-tuning dataset with action tags, tool tags, and optimized responses, and use it to train a series of COLLT models based on five Chinese foundation models: ChatGLM-6B, LLaMA-3-8B, InternLM-3-8B, Qwen-2.5-7B, and Baichuan-2-7B. Experimental results show that COLLT significantly outperforms baselines across nine standard legal NLP tasks. In a free-form QA evaluation based on 500 real-world legal consultation queries, COLLT achieves notable improvements in answer accuracy, legal knowledge coverage, and response appropriateness. Further visualization analysis reveals that COLLT consistently selects appropriate tools under similar intents, and statistical analysis indicates that multi-turn clarification interactions contribute to generating higher-quality responses.

Authors

Institutions

Publication Details

Journal
Mathematics
Published
2026-05-29
DOI
https://doi.org/10.3390/math14111891
Primary Topic
Artificial Intelligence in Law
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

COLLT: A Multi-Task Optimization Framework for Clarification-Oriented Tool Learning in Legal Large Language Models

Zhenxing Wang, Xu Chang, Kaixin Yang, Jingyun Sun
Mathematics
Artificial Intelligence in Law
article

COLLT: A Multi-Task Optimization Framework for Clarification-Oriented Tool Learning in Legal Large Language Models

Zhenxing Wang, Xu Chang, Kaixin Yang, Jingyun Sun
article en

Abstract

Tool-augmented Large Language Models (LLMs) have demonstrated remarkable capabilities in language understanding and generation across various domains, with notable progress in legal applications. However, existing legal LLMs still face two major challenges in legal question answering: (1) a persistent occurrence of hallucinations in generated legal advice and (2) limited effectiveness in handling ambiguous legal queries. To address both issues, this article introduces COLLT (Clarification-Oriented Legal Large language models enhanced by Tool learning), a legal LLM framework designed around a tool learning mechanism oriented to clarification. We model the clarification, tooling, and response workflow as a budgeted sequential decision. All components, including action selection, tool invocation, and response generation, are jointly optimized through multi-task instruction tuning. The COLLT framework first determines whether a user query contains ambiguity. If ambiguity is detected, the model actively guides the user to clarify their intent; if the query is clear, the model automatically invokes appropriate legal tools to improve the accuracy and reliability of its response. To support this mechanism, we propose an end-to-end training strategy enabling the model to learn how to invoke tools adaptively across various scenarios. Specifically, we construct an instruction-tuning dataset with action tags, tool tags, and optimized responses, and use it to train a series of COLLT models based on five Chinese foundation models: ChatGLM-6B, LLaMA-3-8B, InternLM-3-8B, Qwen-2.5-7B, and Baichuan-2-7B. Experimental results show that COLLT significantly outperforms baselines across nine standard legal NLP tasks. In a free-form QA evaluation based on 500 real-world legal consultation queries, COLLT achieves notable improvements in answer accuracy, legal knowledge coverage, and response appropriateness. Further visualization analysis reveals that COLLT consistently selects appropriate tools under similar intents, and statistical analysis indicates that multi-turn clarification interactions contribute to generating higher-quality responses.

MathematicsVol. 14(11)
Northeast Forestry University (CN)
Northeast Forestry University
Peace, Justice and strong institutions
Openalex Percentile: Top 3%
Artificial Intelligence in Law
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.