HKUST(GZ) Logo
HKUST(GZ) · MPhil Thesis Defense · April 2026

Reasoning Capability Enhancement of Multimodal Large Language Models in Specialized Domains

Geometry & Table Understanding
Candidate: Shengen Wu
Student ID: 50029276
Supervisor: Prof. Yutao YUE  |  Co-Supervisor: Prof. Jing TANG
Program: MPhil in Artificial Intelligence · Red Bird
1 / 31

Today's Talk

01
Background & Motivation
Why do MLLMs struggle in specialized domains?
02
Thesis Framework
Three deficiencies → two research lines
03
Part I: Geometry
Multi-view trace synthesis pipeline
04
Part II: Table Reasoning
Table-R1: staged RL with data-side design
05
Results & Analysis
Experiments, ablations, key findings
06
Conclusion
Contributions & future directions
⏱ Estimated time: ~50 minutes + Q&A
2 / 31

MLLMs Are Powerful — But Struggle in Specialized Domains

Models like GPT-4o and Qwen2-VL perform well on general tasks. But in plane geometry and complex table understanding, they still make many mistakes.

📐

Plane Geometry

Requires precise visual perception of diagrams + multi-step symbolic reasoning. One wrong angle reading → wrong answer.

📊

Table Understanding

Tables have merged cells, hierarchical headers, dense numbers. Models must parse structure AND reason logically.

Key insight: The problem is not just the model architecture — it's also the quality of training data.

3 / 31

Three Structural Deficiencies in Training Data

1

Impoverished Supervision Representations

Existing datasets only provide one format — e.g., a symbolic program or a short final answer. This is too narrow for the model to learn from effectively.

→ Representation Enrichment
2

Unfavorable Initialization States

When we apply Reinforcement Learning, the model starts too weak. It can't generate correct answers, so it gets no useful reward signal — training stalls.

→ Initialization Shaping
3

Unbounded Exploration Spaces

For long reasoning chains, the model has to search a huge space of possible steps. Without constraints, it rarely finds the correct path.

→ Exploration Bounding
4 / 31

Two Research Lines, Three Shared Mechanisms

Part I: Geometry
  • Multi-view trace synthesis
  • DSL → PAL + CoT formats
  • Execution-based verification
  • Addresses: ①③
→
Part II: Table (Table-R1)
  • Warm-up data construction
  • Perception-alignment instructions
  • Hint-Completion partitioning
  • Addresses: ①②③

Three shared data-side mechanisms:

① Representation
Enrichment
② Initialization
Shaping
③ Exploration
Bounding

Research question: Can structured data-side interventions systematically improve multimodal reasoning in specialized domains?

5 / 31
Part I

Geometry Reasoning

Multi-View Trace Synthesis for Plane Geometry

Representation Enrichment ①
Exploration Bounding ③
6 / 31

What is Plane Geometry Problem Solving?

Given a geometry diagram + question, the model must reason step-by-step to find the answer.

Example Problem:

📐 Given: AC⊥BD at B, AD⊥CD at D,
AB = 32, DB = 2. Find DC (= y).

Key steps: Geometric Mean Theorem → BD²=AB×CB
Then Pythagorean Theorem in △BDC → DC = y

Problem instance = (D, Q, S, P, A)

D — Diagram image
Q — Natural language question
S — Semantic clauses (e.g., AB∥CD)
P — Gold solution program (DSL)
A — Numeric answer

Dataset: PGPS9K — 9K geometry problems with diagrams

7 / 31

The Problem: DSL is Hard to Learn From

Traditional DSL (Domain-Specific Language):

// Opaque, non-semantic
Equal(LengthOf(Line(A,B)), x)
Equal(LengthOf(Line(B,C)), y)
IsoscelesTriangle(A,B,C)
Find(x)

❌ Hard to verify by running code
❌ Doesn't use model's Python/NL knowledge
❌ Only one rigid format

What we want instead:

✅ Executable — can run and verify automatically
✅ Human-readable — uses natural language
✅ Diverse — multiple formats to learn from
✅ Leverages pre-training — Python, math, NL

Our solution: synthesize 4 complementary formats from the same DSL program

8 / 31

Four Supervision Formats: Geo4

PROGRAM (original)

The raw DSL steps. Kept as one of the four formats to preserve symbolic reasoning style.

PAL (synthesized)

Python + SymPy code. Executable and verifiable. Translated rule-by-rule from DSL.

CoT-Program (synthesized)

Natural language explanation of each DSL step. Written by an LLM, aligned to program steps.

CoT-PAL ⭐ (synthesized)

PAL code + natural language rationale before each equation. Best of both worlds.

Geo3 = PAL + CoT-Program + CoT-PAL  |  Geo4 = Geo3 + Program (32K total training samples)
9 / 31

Synthesis Pipeline: Four Formats from One Problem

Geo data example
10 / 31

Synthesis Pipeline: Three Steps

🔧

Step 1

Program Simplification

Replace variable names with actual numbers.
Make the program self-contained.

→
⚙️

Step 2

Rule-Based PAL Conversion

Each DSL operator → SymPy equation.
Run in sandbox.
Check answer matches gold (ε=0.001).

→
✍️

Step 3

CoT Augmentation

LLM adds natural language rationale to PAL → CoT-PAL.
Also rewrites DSL steps → CoT-Program.

📊 Result: 8,021 programs → 8,017 verified PAL scripts (only 4 filtered out) → 32K Geo4 training set

💡 Key design: Translate-then-Rewrite (rule-based PAL first, then LLM adds CoT) uses only 0.9K tokens per prompt vs. 5.1K for direct generation — and gives better results.

11 / 31

Main Results on PGPS9K

Ours (two-stage + Geo4) vs same-backbone baseline (single-stage):

Backbone Baseline Ours Gain
Qwen2-VL-2B 43.4 50.8 +7.4
Qwen2.5-VL-3B 46.7 58.3 +11.6
Qwen3-VL-8B 58.6 71.2 +12.6

Avg gain across 2B/3B/8B: +10.5 points

Cross-family comparison reported in Chapter 3:

Category Model Acc.
ClosedGPT-4V33.3
ClosedGPT-4o50.9
ClosedClaude-3.7 Sonnet53.7
OpenLLaVA-v1.51.8
OpenQwen2.5-VL-7B-Instruct39.4
SpecializedPGPSNet62.7
SpecializedGF-Reasoner68.7
SpecializedLLaVA-v1.5 + Solver21.6
SpecializedNGS / Geoformer34.1 / 35.6
SpecializedInterGPS (Diag GT)59.8
SpecializedLANS / GeoX66.1 / 52.7
OpenVision-R1-7B50.5
OpenKimi-VL-A3B-Thinking49.2
OursQwen3-VL-8B + Geo471.2

71.2 surpasses GPT-4o (50.9), Claude-3.7 (53.7), PGPSNet (62.7), GF-Reasoner (68.7).

12 / 31

Ablation: Why Multi-View Data Matters

Single-stage training, only changing supervision format:

Training Data 2B 3B 8B Trace Corr.
Program (8K)46.254.267.5--
CoT-Program (8K)46.054.567.996.0%
PAL (8K)46.754.467.4--
CoT-PAL (8K)47.255.468.398.5%
Geo4 (32K, all traces)50.858.371.2--

Construction ablation for CoT-PAL:

Method 2B 3B 8B Prompt Len
Translate-then-Rewrite (ours) 47.2 55.4 68.3 0.9K
Direct LLM: Program → CoT-PAL 43.1 52.6 65.4 5.1K

Geo4 beats strongest single trace (CoT-PAL) by +3.6 / +2.9 / +2.9 on 2B/3B/8B.

Takeaway: data diversity drives gains, and pipeline engineering improves both quality and token efficiency.

13 / 31

Part I: Key Takeaways

✅

Representation Enrichment Works

Transforming opaque DSL into diverse, human-readable, executable formats improves learning. The model can leverage its Python and NL pre-training knowledge.

✅

Diversity Matters More Than Any Single Format

No single trace format is sufficient. Mixing all four formats prevents overfitting to one reasoning style and gives the best results.

✅

Execution-Based Verification Ensures Quality

Running the PAL code in a sandbox and checking the answer filters out wrong traces automatically. 99.95% of programs pass verification.

14 / 31
Part II

Table Reasoning

Table-R1: Data Construction for Multimodal Table Reinforcement Learning

Representation Enrichment ①
Initialization Shaping ②
Exploration Bounding ③
15 / 31

Why Use Reinforcement Learning for Tables?

SFT (Supervised Fine-Tuning)

✅ Easy to train
✅ Good on seen table formats
❌ Imitation learning — just copies examples
❌ Struggles on unseen table structures
❌ Weak on complex multi-step reasoning

→

RL (e.g., GRPO)

✅ Learns from trial and error
✅ Better generalization potential
❌ Needs the model to already be "good enough"
❌ Long reasoning chains → huge search space
❌ Sparse rewards → training can stall

Our approach: fix RL's weaknesses from the data side — don't change the algorithm, change the training environment.

16 / 31

Challenge 1: The Cold-Start Problem

GRPO works by sampling multiple answers, comparing them, and updating the model. But if the model is too weak, all answers are wrong → no useful signal.

Empirical evidence (TabMWP, GRPO):

Model Init Acc. Final Acc. Gain
Qwen2.5-0.5B21.4%28.7%+7.3
Qwen2.5-1.5B31.2%44.0%+12.8
Qwen2.5-3B ⭐55.2%87.6%+32.4
Qwen2.5-7B81.8%91.2%+9.4
Cold-start illustration

Why? Reward variance theory:

If reward ≈ Bernoulli(p), variance = p(1−p)

• p = 0.21 → variance = 0.17 (too low, mostly wrong)
• p = 0.55 → variance = 0.25 ✅ (best signal)
• p = 0.82 → variance = 0.15 (too low, mostly right)

Sweet spot: ~50% initial accuracy

Solution: Use warm-up data to move the model into the "sweet spot" before RL starts.

17 / 31

Table-R1: Three-Stage Framework

Table-R1 framework

Stage 1 Warm-up (SFT) → Stage 2 PA-GRPO (TEDS reward) → Stage 3 HC-GRPO (hint-completion RL) | Datasets: WTQ, TabMWP, HiTab, TabFact, TAT-QA, InfoTabs

18 / 31

Hint-Completion: A Concrete Example

Challenge: For a 5-step reasoning chain, the model must get all 5 steps right. Probability drops exponentially.

Original full solution (5 steps):

Q: In the table, what is the total revenue for Q1 and Q2?
s1: Look at Q1 row → revenue = $120M
s2: Look at Q2 row → revenue = $145M
s3: Add them: 120 + 145 = 265
s4: Unit is million dollars
s5: Answer: $265M
→

HC splits (split at step j=2, j=3):

Split 1 (j=2):
Hint: s1+s2 (Q1=$120M, Q2=$145M)
Model must complete: s3→s4→s5 (just add and answer)
Split 2 (j=3):
Hint: s1+s2+s3 (sum=265)
Model must complete: s4→s5 (just state unit and answer)

Shorter completion → higher chance of success → more useful reward signal for GRPO

19 / 31

Main Results: Full Table from Thesis

Method Res. TabMWP_I WTQ_I HiTab_I TAT-QA_O TabFact_I InfoTabs_O Avg. I Avg. O
Closed-Source MLLM
OpenAI-o4-miniUNK86.7078.2044.8056.1684.7078.3073.6067.23
GPT-4oUNK87.5964.3939.3253.8573.3379.5066.1666.68
Gemini-2.5-ProUNK89.9080.3446.4456.2985.0277.1575.4366.72
Claude-3.5-SonnetUNK83.3071.8041.3159.3360.0870.3064.1264.82
Open-Source MLLM
Qwen2-VL-2BDyn.46.1022.3022.9030.448.9024.6025.0527.52
DeepSeek-VL2 4.5BDyn.53.7540.4218.8924.1113.6524.7931.6824.45
mPLUG-Owl2 7B4486.830.670.130.398.2126.193.9613.29
Monkey 7B89613.2619.076.4112.3122.5622.1115.3317.21
Qwen2-VL-7B-InsDyn.49.5119.735.3320.8540.0046.5628.6433.71
Qwen2-VL-7BDyn.63.8046.5033.1046.507.4032.8037.7039.65
LLaVA-v1.5 7B3366.051.242.032.9718.9028.317.0615.64
Table-LLaVA 7B33657.7818.4310.0912.82*59.8565.26*36.5439.04
Table-LLaVA 13B33659.7720.4110.8515.67*65.0066.91*39.0141.29
Qwen2-VL-72BDyn.81.9565.7044.0452.2373.4572.8266.2962.53
QVQ-72B-PreviewDyn.86.2068.2045.7055.4877.6874.6469.4565.06
Optimized MLLM
Qwen2-VL-2B-SFTDyn.70.0031.0031.9019.6951.7037.4046.1528.55
Qwen2-VL-2B-GRPODyn.71.4035.3035.2029.0222.7029.0041.1529.01
Qwen2-VL-2B-Table-R1 (Ours)Dyn.83.2034.4037.3026.4260.9043.6053.9535.01
Qwen2-VL-7B-SFTDyn.90.3046.8048.5037.8273.2057.6064.7047.71
Qwen2-VL-7B-GRPODyn.89.2053.2054.7051.6811.9041.6052.2546.64
Qwen2-VL-7B-Table-R1 (Ours)Dyn.92.6050.3058.2048.0673.4062.8068.6355.43

7B ours: Avg. In 68.63 (+3.93 vs SFT, +16.38 vs GRPO), Avg. Out 55.43 (+7.72 vs SFT, +8.79 vs GRPO). * denotes trained on corresponding dataset.

20 / 31

Ablation: Each Component Matters

Effect of Warm-up (2B model):

MethodAvg. QATabFact
Table-R151.460.9
w/o Warm-up38.920.5
Drop−12.5−40.4

Without warm-up, TabFact drops 40%! The model can't learn from RL without a good starting point.

Effect of HC splits (2B, TabMWP):

MethodAccuracy
Wqs (no HC)60.2
Wqs + GRPO76.4
Whc (HC warm-up)63.6
Whc + HC-GRPO83.0

HC-GRPO (+6.6 over standard GRPO) — bounding the exploration space makes a real difference.

21 / 31

Part II: Key Takeaways

✅

Warm-up is Essential for RL Stability

Without warm-up, GRPO can completely fail (−40% on TabFact). Moving the model to a "sweet spot" initial accuracy is a prerequisite for effective RL.

✅

Hint-Completion Bounds the Exploration Space

By giving the model a partial reasoning prefix as a "hint", we reduce the complexity of each completion task. More correct rollouts → better GRPO signal.

✅

Data-Side Interventions Generalize

Table-R1 improves not just on seen datasets (held-in) but also on unseen ones (held-out +7.7%), showing that better data design leads to better generalization.

22 / 31

Three Mechanisms Across Both Domains

Mechanism Geometry (Part I) Table (Part II)
① Representation Enrichment DSL → PAL + CoT-PAL + CoT-Program
4 diverse formats
Diverse perception instructions
TEDS continuous reward
② Initialization Shaping — (SFT only, not needed) Warm-up SFT stage
Moves model to "sweet spot"
③ Exploration Bounding Execution-based verification
Filters wrong PAL traces
Hint-Completion partitioning
Bounds rollout horizon

The same three data-side mechanisms appear in both domains — suggesting a general framework for adapting MLLMs to specialized tasks.

23 / 31

Summary of Contributions

① Multi-View Synthesis Paradigm for Geometry

Designed a pipeline to transform opaque DSL into 4 supervision formats. Qwen3-VL-8B: 58.6% → 71.2% on PGPS9K.

② Empirical Characterization of Initialization in Tabular RL

Showed that initial policy accuracy is a key factor for GRPO. Motivated warm-up data construction as a prerequisite for stable RL.

③ Hint-Completion Synthesis for Table Understanding

Introduced trajectory partitioning to bound exploration space. Qwen2-VL-7B: SFT 64.7%/47.7% → Table-R1 68.6%/55.4% (held-in/held-out).

24 / 31

Limitations & Future Directions

Current Limitations:

Geometry: only tested on PGPS9K; other geometry benchmarks not evaluated
Table: gains not uniform across all datasets (e.g., WTQ slightly lower than GRPO)
Data-side focus: architectural improvements not explored

Future Directions:

🔭 Apply multi-view synthesis to other domains (e.g., scientific charts, medical imaging)
🔭 Adaptive HC splitting — learn where to split based on difficulty
🔭 Combine data-side and architecture-side improvements
25 / 31
Conclusion

Data-side interventions provide a practical path to more reliable multimodal reasoning in specialized domains.

✅ Representation enrichment: multi-view supervision improves specialized reasoning versus single-format supervision.

✅ Initialization shaping: warm-up is necessary for stable RL and prevents catastrophic collapse on hard datasets.

✅ Exploration bounding: Hint-Completion partitioning reduces search difficulty and improves long-chain reasoning.

26 / 31

PAL Conversion: How It Works Step by Step

DSL operator types:

34 geometric operators (e.g., IsoscelesTriangle, ParallelLines)
Each maps to a SymPy equation via a fixed rule table.

Operand types:
• N: known values from problem text
• V: intermediate variables to solve
• C: constants (π, 180, etc.)

Verification criteria:

1. Script executes without error (timeout: 30s)
2. All solutions are real numbers
3. All solutions are non-negative (lengths/angles)
4. Answer matches gold within ε = 0.001

Pass rate: 8,017 / 8,021 = 99.95%

27 / 31

How GRPO Works (Simplified)

GRPO = Group Relative Policy Optimization

Step 1: For each question, sample G=4 answers from the model
Step 2: Score each answer (1 if correct, 0 if wrong)
Step 3: Compute advantage = score − group average
Step 4: Update model to increase probability of high-advantage answers

Why it can fail:

If all 4 answers are wrong:
scores = [0, 0, 0, 0]
advantages = [0, 0, 0, 0]
→ No gradient, no learning!

This happens when the model is too weak (cold start) or the task is too hard (long chain).
HC-GRPO fixes this by making each completion task easier → more correct answers in the group.
28 / 31

Dataset Overview

Dataset Task Table Style Sampled Split
WTQFlat TQAWeb8KHeld-in
HiTabHierarchical TQAExcel8KHeld-in
TabMWPNumerical ReasoningWeb8KHeld-in
TabFactFact VerificationExcel/MD8KHeld-in
TAT-QAFinancial ReasoningMarkdown—Held-out
InfoTabsFact VerificationWeb—Held-out

Held-out datasets are never seen during training — used to test generalization.

29 / 31

HC Splits Sensitivity Analysis

How many HC splits per solution should we use? (Qwen2-VL-2B, TabMWP, G=4 rollouts)

HC Splits TabMWP Accuracy
181.5%
282.6%
3 (default)83.2%
483.4%

More splits = more diverse completion problems = better training signal. We use 3 as default: good performance with reasonable data construction cost.

30 / 31
🎓

Thank You

Questions & Discussion

Shengen Wu

MPhil in Artificial Intelligence · HKUST(GZ) · 2026

31 / 31