ChatR1: Reinforcement Learning for Conversational Reasoning and Retrieval Augmented Question Answering
Lupart, Aliannejadi, Kanoulas
We present ChatR1, a reasoning framework based on reinforcement learning (RL) for conversational question answering (CQA). Reasoning plays an important role in CQA, where user intent evolves across dialogue turns, and utterances are often underspecified, requiring contextual interpretation, query reformulation, and dynamic coordination between retrieval and generation. Unlike static `rewrite, retrieve, and generate' pipelines, ChatR1 interleaves search and reasoning across turns, enabling exploratory and adaptive behaviors learned through RL. To address the challenge of sparse and delayed rewards in RL, we propose an intent-aware reward that provides turn-level feedback by aligning retrieval and reasoning with evolving user goals. Our proposed ChatR1 demonstrates strong performance on both 3B and 7B model backbones, outperforming competitive models on five CQA datasets, measured by different metrics (F1, BERTScore, and LLM-as-judge). We include a diverse set of CQA datasets to cover topic shifts, evolving intents, mixed-initiative dialogues, and multi-document grounding, testing ChatR1's performance from various aspects. Ablation studies confirm the effectiveness of the intent-aware reward. Our analyses further reveal diverse reasoning trajectories and effective use of the search tool. ChatR1 also generalizes robustly across domains, demonstrating that RL-based reasoning enables more flexible and context-sensitive behavior than static CQA pipelines.
academic
ChatR1: কথোপকথনমূলক যুক্তিবিদ্যা এবং পুনরুদ্ধার বর্ধিত প্রশ্নোত্তরের জন্য শক্তিশালীকরণ শিক্ষা
এই পেপারটি ChatR1 প্রস্তাব করে, যা শক্তিশালীকরণ শিক্ষার উপর ভিত্তি করে তৈরি একটি কথোপকথনমূলক প্রশ্নোত্তর যুক্তিবিদ্যা কাঠামো। কথোপকথনমূলক প্রশ্নোত্তরে, ব্যবহারকারীর অভিপ্রায় বহু-পর্যায়ের কথোপকথনে ক্রমাগত বিকশিত হয়, বক্তৃতা প্রায়শই অসম্পূর্ণভাবে প্রকাশ করা হয় এবং প্রসঙ্গ ব্যাখ্যা, প্রশ্ন পুনর্নির্মাণ এবং পুনরুদ্ধার-উৎপাদনের গতিশীল সমন্বয়ের প্রয়োজন। স্থির "পুনর্লিখন-পুনরুদ্ধার-উৎপাদন" পাইপলাইনের বিপরীতে, ChatR1 বহু-পর্যায়ের কথোপকথনে অনুসন্ধান এবং যুক্তিবিদ্যা পরিবর্তন করে, শক্তিশালীকরণ শিক্ষার মাধ্যমে অন্বেষণমূলক এবং স্ব-অভিযোজনশীল আচরণ অর্জন করে। শক্তিশালীকরণ শিক্ষায় বিরল এবং বিলম্বিত পুরস্কারের চ্যালেঞ্জ সমাধানের জন্য, লেখকরা অভিপ্রায়-সচেতন পুরস্কার প্রস্তাব করেন, যা পুনরুদ্ধার এবং যুক্তিবিদ্যাকে বিকশিত ব্যবহারকারী লক্ষ্যের সাথে সারিবদ্ধ করে পর্যায়-স্তরের প্রতিক্রিয়া প্রদান করে। ChatR1 3B এবং 7B মডেল উভয়েই চমৎকার কর্মক্ষমতা প্রদর্শন করে, পাঁচটি CQA ডেটাসেটে প্রতিযোগী মডেলগুলিকে অতিক্রম করে।
স্থির পাইপলাইন সীমাবদ্ধতা: বিদ্যমান পদ্ধতিগুলি প্রধানত "পুনর্লিখন-পুনরুদ্ধার-উৎপাদন" স্থির পাইপলাইন ব্যবহার করে, নমনীয়তার অভাব রয়েছে
তত্ত্বাবধানকৃত শিক্ষার উপর নির্ভরতা: বেশিরভাগ পদ্ধতি তত্ত্বাবধানকৃত সূক্ষ্ম-সুরক্ষা (SFT) এর উপর নির্ভর করে, প্রশিক্ষণের সময় অদেখা কথোপকথনের দৃশ্যে খাপ খাইয়ে নেওয়া কঠিন
একক-পর্যায় মিথস্ক্রিয়া অনুমান: বিদ্যমান RL যুক্তিবিদ্যা কাঠামো প্রধানত একক-পর্যায় মিথস্ক্রিয়ার জন্য, বহু-পর্যায়ের কথোপকথনের জটিলতা বিবেচনা করে না
বাণিজ্যিক সিস্টেম (যেমন Perplexity.ai, SearchGPT) ক্রমবর্ধমানভাবে বহু-পর্যায়ের কথোপকথন অনুসন্ধানের দিকে ঝুঁকছে, কিন্তু একাডেমিক গবেষণা এই ক্ষেত্রে তুলনামূলকভাবে পিছিয়ে আছে। শক্তিশালীকরণ শিক্ষা মডেলগুলিকে স্থির প্রদর্শন ডেটার উপর নির্ভর না করে গতিশীল পুনরুদ্ধার এবং যুক্তিবিদ্যা কৌশল শিখতে সক্ষম করতে পারে।
ChatR1 কাঠামো প্রস্তাব: প্রথম RL-ভিত্তিক CQA যুক্তিবিদ্যা মডেল, যা বহু-পর্যায়ের পুনরুদ্ধার এবং উৎপাদন সম্পূর্ণভাবে অপ্টিমাইজ করে, স্থির পাইপলাইনের পরিবর্তে গতিশীল আচরণ শেখে
অভিপ্রায়-সচেতন পুরস্কার ডিজাইন: CQA-এর জন্য বিশেষভাবে ডিজাইন করা পুরস্কার প্রক্রিয়া, বিকশিত ব্যবহারকারী অভিপ্রায়ের সাথে সারিবদ্ধতার মাধ্যমে পুরস্কার বিরলতা হ্রাস করে
ব্যাপক পরীক্ষামূলক যাচাইকরণ: পাঁচটি বিভিন্ন জটিলতার CQA ডেটাসেটে কর্মক্ষমতা যাচাই করে, ক্রস-ডোমেইন সাধারণীকরণ ক্ষমতা প্রদর্শন করে
গভীর বিশ্লেষণাত্মক অন্তর্দৃষ্টি: ChatR1 বৈচিত্র্যময় যুক্তিবিদ্যা পথ উৎপাদন করে, কার্যকরভাবে অনুসন্ধান সরঞ্জাম ব্যবহার করে এবং ক্রস-ডোমেইন শক্তিশালীতা প্রদর্শন করে তা প্রকাশ করে
ডেটাসেট D দেওয়া হয়েছে যা বহু-পর্যায়ের ব্যবহারকারী-সিস্টেম কথোপকথন ধারণ করে, প্রতিটি কথোপকথন একাধিক পর্যায় নিয়ে গঠিত, এবং নথি সংগ্রহ C। প্রতিটি পর্যায়ে, সিস্টেম কথোপকথনের ইতিহাস H এবং বর্তমান ব্যবহারকারী প্রশ্ন q গ্রহণ করে, কাজটি হল একটি উত্তর y উৎপাদন করা, H এর প্রসঙ্গ ব্যবহার করে এবং C এর উপর ভিত্তি করে তথ্যগত যাচাইকরণ করা। ব্যবহারকারী অভিপ্রায় পুনর্লিখিত প্রশ্ন q_rw হিসাবে সংজ্ঞায়িত করা হয়, যা q-তে প্রসঙ্গ উল্লেখ এবং অস্পষ্টতা সমাধান করে।
সম্পূর্ণ অপ্টিমাইজেশন: ঐতিহ্যবাহী বিচ্ছিন্ন পাইপলাইনের বিপরীতে, ChatR1 যুক্তিবিদ্যা, পুনরুদ্ধার এবং উৎপাদন যৌথভাবে অপ্টিমাইজ করে
অভিপ্রায়-সচেতন ডিজাইন: CQA বৈশিষ্ট্যের জন্য বিশেষভাবে ডিজাইন করা পুরস্কার প্রক্রিয়া, পুনরুদ্ধার ফলাফলের উপর নির্ভর না করে সরাসরি প্রশ্ন গুণমান মূল্যায়ন করে
স্ব-অভিযোজনশীল যুক্তিবিদ্যা: RL এর মাধ্যমে কখন এবং কীভাবে অনুসন্ধান করতে হবে তা শিখে, পূর্বনির্ধারিত স্থির কৌশলের পরিবর্তে
ঐতিহ্যবাহী CQA পদ্ধতি প্রধানত স্থির RAG পাইপলাইন এবং তত্ত্বাবধানকৃত সূক্ষ্ম-সুরক্ষার উপর নির্ভর করে, স্পষ্ট যুক্তিবিদ্যা প্রক্রিয়া কখন এবং কীভাবে অনুসন্ধান করতে হবে তা সিদ্ধান্ত নিতে অভাব রয়েছে।
পেপারটি শক্তিশালীকরণ শিক্ষা, কথোপকথন সিস্টেম, তথ্য পুনরুদ্ধার ইত্যাদি ক্ষেত্রের গুরুত্বপূর্ণ কাজ উদ্ধৃত করে, বিশেষত:
PPO অ্যালগরিদম (Schulman et al., 2017)
Search-R1 ইত্যাদি RL যুক্তিবিদ্যা কাজ (Jin et al., 2025)
কথোপকথন প্রশ্নোত্তর ডেটাসেট নির্মাণ কাজ (Adlakha et al., 2022; Anantha et al., 2021)
সামগ্রিক মূল্যায়ন: এটি একটি উচ্চ-মানের গবেষণা পেপার, প্রযুক্তিগত উদ্ভাবন, পরীক্ষামূলক ডিজাইন এবং বিশ্লেষণাত্মক গভীরতার দিক থেকে চমৎকার কর্মক্ষমতা প্রদর্শন করে। শক্তিশালীকরণ শিক্ষা বহু-পর্যায়ের কথোপকথন প্রশ্নোত্তরে প্রবর্তন করা একটি অর্থপূর্ণ গবেষণা দিক, এবং অভিপ্রায়-সচেতন পুরস্কারের ডিজাইন চতুরতার সাথে CQA-তে মূল চ্যালেঞ্জ সমাধান করে। কিছু সীমাবদ্ধতা থাকলেও, পেপারটি ক্ষেত্রে গুরুত্বপূর্ণ অবদান প্রদান করে, আরও গবেষণা এবং প্রয়োগের যোগ্য।