2025-11-18T01:52:13.530679

Mulberry: Empowering MLLM with o1-like Reasoning and Reflection via Collective Monte Carlo Tree Search

Yao, Huang, Wu et al.
In this work, we aim to develop an MLLM that understands and solves questions by learning to create each intermediate step of the reasoning involved till the final answer. To this end, we propose Collective Monte Carlo Tree Search (CoMCTS), a new learning-to-reason method for MLLMs, which introduces the concept of collective learning into ``tree search'' for effective and efficient reasoning-path searching and learning. The core idea of CoMCTS is to leverage collective knowledge from multiple models to collaboratively conjecture, search and identify effective reasoning paths toward correct answers via four iterative operations including Expansion, Simulation and Error Positioning, Backpropagation, and Selection. Using CoMCTS, we construct Mulberry-260k, a multimodal dataset with a tree of rich, explicit and well-defined reasoning nodes for each question. With Mulberry-260k, we perform collective SFT to train our model, Mulberry, a series of MLLMs with o1-like step-by-step Reasoning and Reflection capabilities. Extensive experiments demonstrate the superiority of our proposed methods on various benchmarks. Code will be available at https://github.com/HJYao00/Mulberry
academic

মালবেরি: সম্মিলিত মন্টে কার্লো ট্রি সার্চের মাধ্যমে MLLM কে o1-সদৃশ যুক্তি এবং প্রতিফলন ক্ষমতা প্রদান করা

মৌলিক তথ্য

  • পেপার আইডি: 2412.18319
  • শিরোনাম: মালবেরি: সম্মিলিত মন্টে কার্লো ট্রি সার্চের মাধ্যমে MLLM কে o1-সদৃশ যুক্তি এবং প্রতিফলন ক্ষমতা প্রদান করা
  • লেখক: হুয়ান্জিন ইয়াও, জিয়াক্সিং হুয়াং, ওয়েনহাও উ, জিংগিয়ি ঝাং, ইবো ওয়াং, শুনিউ লিউ, ইংজিয়ে ওয়াং, ইউক্সিন সং, হাওচেং ফেং, লি শেন, ডাচেং তাও
  • শ্রেণীবিভাগ: cs.CV cs.AI
  • প্রকাশনার সময়: ২০২৪ সালের ৩১ ডিসেম্বর (arXiv v2)
  • পেপার লিংক: https://arxiv.org/abs/2412.18319

সারসংক্ষেপ

এই গবেষণার লক্ষ্য এমন একটি বহুমোডাল বৃহৎ ভাষা মডেল (MLLM) তৈরি করা যা যুক্তি প্রক্রিয়ার প্রতিটি মধ্যবর্তী ধাপ তৈরি করে শিখে সমস্যা বুঝতে এবং সমাধান করতে পারে। এই উদ্দেশ্যে, লেখকরা সম্মিলিত মন্টে কার্লো ট্রি সার্চ (CoMCTS) প্রস্তাব করেছেন, যা MLLM শেখার জন্য একটি নতুন পদ্ধতি যা সম্মিলিত শেখার ধারণাকে "ট্রি সার্চ"-এ প্রবর্তন করে, কার্যকর এবং দক্ষ যুক্তি পথ অনুসন্ধান এবং শেখা সক্ষম করে। CoMCTS-এর মূল ধারণা হল একাধিক মডেলের সম্মিলিত জ্ঞান ব্যবহার করা, সম্প্রসারণ, অনুকরণ এবং ত্রুটি অবস্থান, পশ্চাদপদ প্রচার এবং নির্বাচন চারটি পুনরাবৃত্তিমূলক ক্রিয়াকলাপের মাধ্যমে সহযোগিতামূলকভাবে অনুমান, অনুসন্ধান এবং সঠিক উত্তরের দিকে কার্যকর যুক্তি পথ চিহ্নিত করা। CoMCTS-এর উপর ভিত্তি করে, লেখকরা মালবেরি-260k ডেটাসেট তৈরি করেছেন এবং o1-সদৃশ ধাপে ধাপে যুক্তি এবং প্রতিফলন ক্ষমতা সহ মালবেরি মডেল সিরিজ প্রশিক্ষণ দিয়েছেন।

গবেষণা পটভূমি এবং প্রেরণা

সমস্যা সংজ্ঞা

বর্তমান বহুমোডাল বৃহৎ ভাষা মডেল (MLLM) জটিল যুক্তি কাজ পরিচালনা করার সময় ব্যর্থতার হার উল্লেখযোগ্যভাবে বৃদ্ধি পায়। বিদ্যমান MLLM প্রধানত "সরাসরি পূর্বাভাস" মোড গ্রহণ করে, অর্থাৎ প্রশ্নের জন্য সংক্ষিপ্ত চূড়ান্ত উত্তর তৈরি করে, স্পষ্ট এবং সুসংজ্ঞায়িত মধ্যবর্তী যুক্তি পদক্ষেপের অভাব রয়েছে।

গুরুত্ব

ফাইনম্যানের মতে: "আমি যা তৈরি করতে পারি না, আমি তা বুঝতে পারি না।" MLLM-গুলি যুক্তি প্রক্রিয়ার প্রতিটি ধাপ তৈরি এবং গভীরভাবে বুঝতে সক্ষম হওয়া উচিত, যা জটিল কাজ সমাধানের জন্য অত্যন্ত গুরুত্বপূর্ণ।

বিদ্যমান পদ্ধতির সীমাবদ্ধতা

  1. অনুসন্ধান কার্যকারিতা সমস্যা: ঐতিহ্যবাহী MCTS পদ্ধতি স্ব-নির্দেশিত কাজ করে, কিন্তু বর্তমান MLLM স্পষ্ট মধ্যবর্তী যুক্তি পদক্ষেপ প্রশিক্ষণের অভাব রয়েছে, একক MLLM যুক্তি স্থানের মধ্যে সমজাতীয় নিম্নমানের নোডে আটকে যাওয়ার প্রবণতা রয়েছে
  2. অনুসন্ধান দক্ষতা সমস্যা: ঐতিহ্যবাহী MCTS প্রতিটি অনুসন্ধান পুনরাবৃত্তিতে শুধুমাত্র একটি পরবর্তী যুক্তি নোড সম্প্রসারিত করে, অসংখ্য পুনরাবৃত্তির প্রয়োজন, গণনা-নিবিড় MLLM-এর জন্য অদক্ষ

গবেষণা প্রেরণা

OpenAI o1 এবং অন্যান্য সর্বশেষ অগ্রগতি দ্বারা অনুপ্রাণিত, লেখকরা MLLM-এ "ট্রি সার্চ" পদ্ধতি প্রয়োগ করতে চেয়েছিলেন, কিন্তু সরাসরি প্রয়োগ কার্যকর ছিল না, তাই অনুসন্ধান চ্যালেঞ্জ সমাধানের জন্য নতুন সম্মিলিত শেখার প্রক্রিয়া ডিজাইন করার প্রয়োজন ছিল।

মূল অবদান

  1. CoMCTS পদ্ধতি প্রস্তাব: প্রথমবারের মতো সম্মিলিত শেখার ধারণা MCTS-এ প্রবর্তন করা, সম্মিলিত জ্ঞান ব্যবহার করে MLLM-এর কার্যকর এবং প্রতিফলনশীল যুক্তি পথ সহযোগিতামূলকভাবে অনুমান, অনুসন্ধান এবং চিহ্নিত করা
  2. মালবেরি-260k ডেটাসেট নির্মাণ: MLLM ধাপে ধাপে যুক্তি এবং প্রতিফলন গবেষণা অগ্রসর করার জন্য মূল্যবান সম্পদ প্রদান করা
  3. মালবেরি মডেল সিরিজ উন্নয়ন: চমৎকার ধাপে ধাপে যুক্তি এবং প্রতিফলন ক্ষমতা সহ MLLM
  4. পরীক্ষামূলক যাচাইকরণ: একাধিক বেঞ্চমার্ক পরীক্ষায় পদ্ধতির উৎকর্ষতা প্রমাণ করা

পদ্ধতি বিস্তারিত

কাজের সংজ্ঞা

বহুমোডাল ইনপুট প্রশ্ন Q দেওয়া (যেমন ছবি সহ পাঠ্য কাজের নির্দেশনা), লক্ষ্য হল মধ্যবর্তী যুক্তি অবস্থার একটি সিরিজ (s₁, s₂, s₃, ..., sₘ) তৈরি করা, চূড়ান্তভাবে সঠিক উত্তর পাওয়া।

CoMCTS মূল স্থাপত্য

CoMCTS MLLM {π₁, π₂, ..., πₖ} এর একটি সেটের সম্মিলিত জ্ঞান ব্যবহার করে, চারটি মূল ক্রিয়াকলাপের মাধ্যমে যুক্তি পথ পুনরাবৃত্তিমূলকভাবে অনুসন্ধান করে:

(ক) সম্প্রসারণ (Expansion)

বর্তমান পাতার নোড sₖₘ থেকে শুরু করে, একাধিক MLLM ব্যবহার করে বৈচিত্র্যময় এবং পরিপূরক প্রার্থী যুক্তি পথ সমান্তরালভাবে সম্প্রসারিত করা:

S^j_candidate ~ πⱼ(·|Q, Parent(sₖₘ), sₖₘ)

(খ) অনুকরণ এবং ত্রুটি অবস্থান (Simulation and Error Positioning)

প্রার্থী নোড মূল্য মূল্যায়ন করতে সম্মিলিত জ্ঞান ব্যবহার করা, ত্রুটিপূর্ণ যুক্তি নোড চিহ্নিত এবং ফিল্টার করা:

R(sʲᵢ) = (1/K) ∑ᵏₗ₌₁ πₗ(·|prompt_eval, Q, Parent(sʲᵢ), sʲᵢ)
S*_candidate = {sʲᵢ ∈ S_candidate | R(sʲᵢ) ≥ t}

(গ) পশ্চাদপদ প্রচার (Backpropagation)

যুক্তি গাছের প্রতিটি নোডের পরিদর্শন সংখ্যা N এবং নোড মূল্য V নিচ থেকে উপরে আপডেট করা:

V(s) ← [N(s)·V(s) + ∑_{sₗ∈Child(s)} R(sₗ)] / [N(s) + CountChild(S*_candidate, s)]
N(s) ← N(s) + CountChild(S*_candidate, s)

(ঘ) নির্বাচন (Selection)

পরবর্তী শুরু নোড নির্বাচন করতে Upper Confidence Bound (UCB) ব্যবহার করা:

sₖ*ₘ = argmax_{s∈S*_candidate} V(s) + c·√[log N(ŝ)/(1+N(s))]

প্রতিফলনশীল যুক্তি সম্প্রসারণ

CoMCTS দ্বারা নির্মিত একীভূত যুক্তি গাছের উপর ভিত্তি করে, নেতিবাচক নমুনা ভাই-বোন নোড চিহ্নিত করা এবং প্রতিফলনশীল যুক্তি পথ নির্মাণ করা:

  1. নেতিবাচক নমুনা ভাই-বোন নোড চিহ্নিতকরণ:
s_neg = argmin_{sₗ∈Sibling(s)} UCB(sₗ) - UCB(s)
  1. প্রতিফলনশীল যুক্তি পথ নির্মাণ:
Y_reflect = Replace(Y, s, (s_neg, prompt_reflect, s))

সম্মিলিত তদারকি সূক্ষ্ম-সুর (CoSFT)

CoMCTS অনুসন্ধান দ্বারা ডেটা ব্যবহার করে মডেল প্রশিক্ষণ:

L_CoSFT(πₖ) = ∑_{(Q,Y)∈D} log πₖ(Y|Q)
L_CoSFT-Re(πₖ) = ∑_{(Q,Y_reflect)∈D} log πₖ(Y_reflect|Q)

পরীক্ষামূলক সেটআপ

ডেটাসেট

মালবেরি-260k ডেটাসেট গঠন:

  • 55K গণিত ডেটা (GLLaVA, GEOS, UniGeo ইত্যাদি)
  • 116K চার্ট বোঝাপড়া ডেটা (DVQA, DocVQA, ChartQA ইত্যাদি)
  • 41K গণিত প্রয়োগ সমস্যা ডেটা (IconQA, TabMWP, CLEVR ইত্যাদি)
  • 2K চিকিৎসা ডেটা (VQA-RAD, PMC-VQA)
  • 17K বিজ্ঞান ডেটা (TQA, AI2D, ScienceQA)
  • 24K প্রাকৃতিক বিশ্ব প্রশ্নোত্তর ডেটা (VQA-AS, A-OKVQA ইত্যাদি)

মূল্যায়ন মেট্রিক্স

8টি ব্যাপকভাবে ব্যবহৃত চ্যালেঞ্জিং ডেটাসেটে মূল্যায়ন: MathVista, MMStar, MMMU, ChartQA, DynaMath, HallBench, MM-Math, MME

তুলনামূলক পদ্ধতি

  • বন্ধ-উৎস মডেল: GPT-4o, Claude-3.5 Sonnet
  • খোলা-উৎস মডেল: DeepSeek-VL, InternVL2, MiniCPM-V ইত্যাদি
  • যুক্তি মডেল: LLaVA-CoT, LLaVA-Reasoner, Insight-V

বাস্তবায়ন বিবরণ

  • সম্মিলিত শেখা 4টি মডেল ব্যবহার করে: GPT-4o, Qwen2-VL-7B, LLaMA-3.2-11B-Vision-Instruct, Qwen2-VL-72B
  • সর্বোচ্চ অনুসন্ধান পুনরাবৃত্তি সংখ্যা: 20
  • ব্যাচ আকার: 128, শেখার হার: 1e-5, প্রশিক্ষণ যুগ: 2

পরীক্ষামূলক ফলাফল

প্রধান ফলাফল

বেসলাইন মডেলের সাথে তুলনা:

  • মালবেরি-7B Qwen2-VL-7B এর তুলনায় গড়ে 4.2% উন্নতি
  • মালবেরি-11B LLaMA-3.2-11B-Vision-Instruct এর তুলনায় গড়ে 7.5% উন্নতি
  • মালবেরি-2B Qwen2-VL-2B এর তুলনায় গড়ে 5.4% উন্নতি
  • মালবেরি-8B LLaVA-NeXT-8B এর তুলনায় গড়ে 11.0% উন্নতি

যুক্তি প্রতিক্রিয়া মডেলের সাথে তুলনা:

  • MathVista-তে, মালবেরি LLaVA-Reasoner-8B এবং Insight-V-8B এর তুলনায় যথাক্রমে 5.7% এবং 6.5% উন্নতি
  • MMMU-তে যথাক্রমে 3.0% এবং 1.0% উন্নতি

SOTA মডেলের সাথে তুলনা: মালবেরি বেশিরভাগ খোলা-উৎস MLLM-এ সর্বোত্তম পারফরম্যান্স প্রদর্শন করে, কিছু মেট্রিক্সে বন্ধ-উৎস মডেলের কর্মক্ষমতার কাছাকাছি।

বিলোপন পরীক্ষা

CoMCTS উপাদান বিশ্লেষণ (সারণী 2):

  • GPT-4o সরাসরি পূর্বাভাস: 58.2% অনুসন্ধান সাফল্যের হার
  • শুধুমাত্র GPT-4o এর CoMCTS: 63.8%
  • ক্রমান্বয়ে আরও মডেল যোগ করা ক্রমাগত কর্মক্ষমতা উন্নত করে
  • সম্পূর্ণ CoMCTS: 80.2% অনুসন্ধান সাফল্যের হার

প্রতিফলন ডেটা অবদান (সারণী 3): MathVista-তে, প্রতিফলন ডেটা যোগ করা 0.8% কর্মক্ষমতা উন্নতি করে, কার্যকারিতা এবং প্রতিফলনশীল যুক্তি ডেটার পরিপূরকতা প্রমাণ করে।

ট্রি সার্চ পদ্ধতি তুলনা

CoMCTS অন্যান্য ট্রি সার্চ পদ্ধতির তুলনায় উল্লেখযোগ্যভাবে উৎকৃষ্ট:

  • অনুসন্ধান সাফল্যের হার: 80.2% বনাম 66.2% (Omega-MCTS)
  • গড় অনুসন্ধান পুনরাবৃত্তি: 12.7 বনাম 24.3 (Omega-MCTS)

কেস বিশ্লেষণ

গুণগত তুলনা দেখায় যে মালবেরি সমৃদ্ধ, স্পষ্ট এবং সুসংজ্ঞায়িত যুক্তি পদক্ষেপ তৈরি করতে পারে, যখন বেসলাইন মডেল তুলনামূলকভাবে সংক্ষিপ্ত পূর্বাভাস তৈরি করে এবং সহজেই ত্রুটি করে।

সম্পর্কিত কাজ

বহুমোডাল বৃহৎ ভাষা মডেল

MLLM সাধারণ দৃষ্টি-ভাষা বোঝাপড়ায় উল্লেখযোগ্য অগ্রগতি অর্জন করেছে, কিন্তু গভীর যুক্তির প্রয়োজনীয় জটিল কাজে এখনও চ্যালেঞ্জ রয়েছে।

বৃহৎ ভাষা মডেল যুক্তি

যুক্তি পদ্ধতি তিনটি বিভাগে বিভক্ত:

  1. প্রম্পট-ভিত্তিক পদ্ধতি: যেমন Chain-of-Thought (CoT)
  2. পরিকল্পনা-ভিত্তিক পদ্ধতি: যেমন Tree/Graph-of-thought
  3. শেখা-ভিত্তিক পদ্ধতি: যেমন GPT-o1, Star, Iter-MCTS ইত্যাদি

মন্টে কার্লো ট্রি সার্চ

MCTS খেলা, রোবোটিক্স, উপপাদ্য প্রমাণ এবং অন্যান্য ক্ষেত্রে ব্যাপক প্রয়োগ পেয়েছে, এই পেপার প্রথমবারের মতো সম্মিলিত শেখা MCTS-এ MLLM-এর জন্য প্রবর্তন করে।

সম্মিলিত শেখা

সম্মিলিত শেখা একাধিক ব্যক্তির সম্মিলিত বুদ্ধিমত্তা ব্যবহার করে শেখার ফলাফল উন্নত করার লক্ষ্য রাখে, এই পেপার এটি যুক্তি পথ অনুসন্ধানে প্রয়োগ করে।

উপসংহার এবং আলোচনা

প্রধান উপসংহার

  1. CoMCTS সম্মিলিত শেখার মাধ্যমে MLLM-এ ঐতিহ্যবাহী MCTS-এর অনুসন্ধান দক্ষতা এবং কার্যকারিতা সমস্যা কার্যকরভাবে সমাধান করে
  2. মালবেরি-260k ডেটাসেট বহুমোডাল যুক্তি গবেষণার জন্য মূল্যবান সম্পদ প্রদান করে
  3. মালবেরি মডেল একাধিক বেঞ্চমার্ক পরীক্ষায় চমৎকার ধাপে ধাপে যুক্তি এবং প্রতিফলন ক্ষমতা প্রদর্শন করে

সীমাবদ্ধতা

  1. গণনা খরচ: একাধিক মডেল সম্মিলিত অনুসন্ধানে অংশগ্রহণের প্রয়োজন, গণনা ওভারহেড উল্লেখযোগ্য
  2. মডেল নির্ভরতা: অনুসন্ধান গুণমান সম্মিলিত শেখায় অংশগ্রহণকারী মডেলের গুণমানের উপর নির্ভর করে
  3. ডোমেইন অভিযোজনযোগ্যতা: নির্দিষ্ট ডোমেইনে কর্মক্ষমতা প্রশিক্ষণ ডেটা বিতরণ দ্বারা সীমাবদ্ধ হতে পারে

ভবিষ্যত দিকনির্দেশনা

  1. আরও দক্ষ সম্মিলিত শেখার প্রক্রিয়া অন্বেষণ করা
  2. আরও মোডালিটি এবং কাজের ধরনে সম্প্রসারণ করা
  3. স্ব-অভিযোজনশীল যুক্তি পদক্ষেপ বরাদ্দ কৌশল গবেষণা করা

গভীর মূল্যায়ন

শক্তি

  1. পদ্ধতি উদ্ভাবন শক্তিশালী: প্রথমবারের মতো সম্মিলিত শেখা MCTS-এ MLLM-এর জন্য প্রবর্তন করা, ঐতিহ্যবাহী পদ্ধতির মূল সমস্যা সমাধান করা
  2. পরীক্ষা ব্যাপক: একাধিক ডেটাসেট এবং মডেলে সম্পূর্ণ মূল্যায়ন, বিলোপন পরীক্ষা এবং তুলনামূলক বিশ্লেষণ অন্তর্ভুক্ত
  3. ব্যবহারিক মূল্য উচ্চ: নির্মিত ডেটাসেট এবং মডেল সম্প্রদায়ের জন্য গুরুত্বপূর্ণ মূল্য রয়েছে
  4. প্রযুক্তিগত বিবরণ সম্পূর্ণ: পদ্ধতি বর্ণনা স্পষ্ট, বাস্তবায়ন বিবরণ পর্যাপ্ত

অপূর্ণতা

  1. গণনা দক্ষতা: যদিও ঐতিহ্যবাহী MCTS এর তুলনায় উন্নতি হয়েছে, তবুও একাধিক মডেল সহযোগিতার প্রয়োজন, গণনা খরচ উল্লেখযোগ্য
  2. সাধারণীকরণ ক্ষমতা: প্রধানত গণিত এবং চার্ট বোঝাপড়া কাজে যাচাই করা হয়েছে, অন্যান্য জটিল যুক্তি কাজে কর্মক্ষমতা আরও যাচাইকরণের প্রয়োজন
  3. তাত্ত্বিক বিশ্লেষণ অপর্যাপ্ত: সম্মিলিত শেখা কেন কার্যকর তার গভীর তাত্ত্বিক বিশ্লেষণের অভাব

প্রভাব

  1. একাডেমিক অবদান: বহুমোডাল যুক্তি এবং ট্রি সার্চ পদ্ধতির জন্য নতুন গবেষণা দিকনির্দেশনা প্রদান করে
  2. ব্যবহারিক মূল্য: মালবেরি-260k ডেটাসেট এবং মডেল সম্পর্কিত গবেষণা উন্নয়ন প্রচার করতে পারে
  3. পুনরুৎপাদনযোগ্যতা: লেখকরা কোড খোলা-উৎস করার প্রতিশ্রুতি দিয়েছেন, পদ্ধতি প্রচারের জন্য অনুকূল

প্রযোজ্য পরিস্থিতি

  1. গণিত যুক্তি কাজ: বিশেষত বহু-পদক্ষেপ যুক্তির প্রয়োজনীয় গণিত সমস্যার জন্য উপযুক্ত
  2. চার্ট বোঝাপড়া: চার্ট বিশ্লেষণ এবং ডেটা ভিজ্যুয়ালাইজেশন বোঝাপড়ায় চমৎকার কর্মক্ষমতা
  3. বৈজ্ঞানিক প্রশ্নোত্তর: ধাপে ধাপে বিশ্লেষণের প্রয়োজনীয় বৈজ্ঞানিক প্রশ্ন সমাধানের জন্য প্রযোজ্য
  4. শিক্ষা প্রয়োগ: যুক্তি ক্ষমতা সহ শিক্ষামূলক AI সিস্টেম নির্মাণে ব্যবহার করা যেতে পারে

রেফারেন্স

পেপার ব্যাপক সম্পর্কিত কাজ উদ্ধৃত করেছে, যার মধ্যে রয়েছে:

  • বহুমোডাল বৃহৎ ভাষা মডেল: LLaVA, Qwen2-VL ইত্যাদি
  • যুক্তি পদ্ধতি: Chain-of-Thought, Tree-of-Thought ইত্যাদি
  • মন্টে কার্লো ট্রি সার্চ: AlphaGo, MCTS ভেরিয়েন্ট ইত্যাদি
  • সম্মিলিত শেখা: Co-training সম্পর্কিত কাজ ইত্যাদি