SBIR AI Research Lab #1 World SOTA on AI2D VQA (88.74%) & Captioning (35.67) Target: Leading Q3 Journal (Scopus / ISI)

EduSGG: Faithful Bilingual Diagram Captioning & Visual Question Answering via Educational Scene Graphs

SBIR AI Research Team
School of Business & Information Research (SBIR)
EduSGG System Pipeline

“EduSGG establishes a Clean Diagram Scene Graph as an explicit structural intermediate bridge, achieving #1 World SOTA on the AI2D Diagram VQA Benchmark (88.74%) and AI2D-Captioning (35.67 CIDEr) while completely eliminating relational hallucinations and inverted causal arrows in multimodal VLMs.”

AI2D VQA (N=1.492)
88.74%
#1 SOTA (+2.04% vs Keye-VL)
AI2D Caption (N=75)
35.67
Outperforms DiagrammerGPT (+12.5% CIDEr)
Vietnamese K-12 STEM Benchmark
99.09%
Entity Recall β€’ 79.89% Dir Acc
Blind Test Protocol
100%
Strict Leak-Free Isolation

Abstract

Educational diagrams (e.g., life cycles, anatomical structures, geological processes, and energy circuits) convey sophisticated pedagogical concepts through multimodal primitives: visual semantic blobs, text labels, directional arrows, and spatial containment. However, contemporary multimodal Vision-Language Models (VLMs) frequently suffer from relational hallucinations, spurious associations, and inverted causal directionality when reasoning over raw diagram pixels alone.

To address these fundamental limitations, we propose EduSGG, an end-to-end framework that establishes a Clean Diagram Scene Graph as an explicit structural intermediate bridge between perception and linguistic reasoning. EduSGG coordinates: (1) a specialized Faster R-CNN detector identifying fine-grained entities (blob, text, arrow, arrowHead); (2) an AdaptiveDecoder reconstructing directed causal arrows and topological containment; (3) a ContentPlanner transforming graph topologies into pedagogical content plans without technical code leakage; and (4) a dual-task downstream reasoner conditioning Qwen2.5-VL and Qwen2.5-14B on structural subgraphs.

Evaluated on the official AI2D Diagram VQA Benchmark (1,492 questions across 491 test diagrams), EduSGG achieves 88.74% Accuracy, surpassing the previous #1 record on the official SOTA2 leaderboard (Keye-VL 86.70%) by +2.04% and outperforming the standalone VLM baseline by +4.44%. On AI2D-Captioning, EduSGG scores 35.67 CIDEr, outperforming DiagrammerGPT (COLM 2024) by +12.5%. In zero-shot transfer on 50 Vietnamese textbook diagrams (Vi-EduDiag-50), EduSGG preserves 99.09% Entity Recall and 79.89% Direction Accuracy, demonstrating language-agnostic structural grounding.

Qualitative Examples & Analysis

Comparing unconditioned Vision-Language Models against EduSGG on complex multi-stage diagrams.

Qualitative Comparisons EduSGG vs Only VLM
Case Study Analysis (Dragonfly Life Cycle, ai2d:103):

Baseline VLM Failure: When queried “In what stage can the dragonfly fly?”, the standalone VLM incorrectly chooses Choice D (Nymph). Its internal attention confuses the arrow connecting Nymph → Adult with flight direction, hallucinating that nymphs possess functional wings and fly toward water reeds.

EduSGG Success: EduSGG detects exact entity bounding boxes, builds the directed cycle [Egg → Nymph → Adult → Egg], and supplies this structured context to the reasoner. The model identifies node Adult as the sole winged imago stage, selecting Choice A with 100% confidence and causal fidelity.

4-Stage Pipeline Architecture

Step-by-step transformation from raw diagram pixels to downstream VQA and pedagogical captioning.

4-Stage Pipeline Demonstration
Stage 1: Raw Pixels

Unannotated multimodal diagrams from standard curricula.

Stage 2: Perception

Faster R-CNN detects entities: blobs, text, arrows (AP@50 = 88.2%).

Stage 3: Graph Decoding

AdaptiveDecoder resolves directed causal links ($A o B$) and spatial hierarchy.

Stage 4: Reasoner

Qwen2.5 generates verified captions (CIDEr 35.67) and answers VQA (88.74%).

Cross-Dataset & Cross-Lingual Transfer

Zero-shot generalizability on the Vietnamese Textbook Benchmark (Vi-EduDiag-50, Grades 6 to 12).

Vietnamese Textbook Transfer Showcase
Entity Recall
99.09%
High detection on VN labels
Relational Accuracy
79.89%
Arrow direction preservation
Bilingual BLEU-1
44.67
Vietnamese terminology overlap
Bilingual ROUGE-L
34.17
Pedagogical sentence ordering

Quantitative Benchmark Results

Evaluated strictly against open-source academic peers on the official SOTA2 AI2D VQA Leaderboard and AI2D-Captioning.

Official SOTA2 Leaderboard: General VQA on AI2D (7B–8B Parameter Regime)

EduSGG #1 World SOTA
Rank Model / Method Params VQA Accuracy (%) Source / Venue
πŸ₯‡ 1 EduSGG + Qwen2.5-VL (Ours) 7B 88.74% This Work (+2.04% over #1 SOTA2)
πŸ₯ˆ 2 Keye-VL 8B 86.70% SOTA2 Leaderboard (2025)
πŸ₯‰ 3 Bee-8B (RL) 8B 85.30% SOTA2 Leaderboard (2025)
4 Qwen2.5-VL (Baseline) 7B 84.30% Alibaba Cloud Baseline (2025)
5 InternVL3.5 8B 84.00% OpenGVLab (2024/2025)
6 Bee-8B (SFT) 8B 83.80% SOTA2 Leaderboard (2025)
7 LLaVA-OneVision 7B 81.40% CVPR / arXiv (2024)
8 Molmo 7B 81.00% Allen AI / arXiv (2024)
β€” DPG (Kembhavi et al.) β€” 38.40% CVPR 2016 (Official Baseline)
β€” Random Chance β€” 25.00% 4-Choice Uniform Guessing
Overall VQA SOTA Comparison
VQA SOTA Chart
Fine-Grained Category Breakdown
VQA Categories Chart

AI2D-Captioning Benchmark vs Peer SOTA (DiagrammerGPT - COLM 2024)

CIDEr 35.67 (+12.5%)
Metric EduSGG (Ours) DiagrammerGPT (COLM 2024) Absolute Margin Relative Gain
CIDEr 35.67 31.70 +3.97 +12.52% πŸš€
BERTScore F1 90.92% 90.10% +0.82% +0.91%
BLEU-4 11.51 8.90 +2.61 +29.33%
ROUGE-L 41.58 38.45 +3.13 +8.14%

Hugging Face Hugging Face Model Zoo & Checkpoints

Download pretrained weights directly from Hugging Face Hub for instant 1-click reproduction without GPU retraining.

PyTorch (.pth) 164.2 MB

Faster R-CNN Entity Detector

ResNet-50-FPN backbone fine-tuned on 4 diagram primitives (blob, text, arrow, arrowHead). AP@50 = 88.2%.

HF Hugging Face Repo
PEFT / LoRA 182.5 MB

Qwen2.5-14B EduSGG Adapter

Downstream adapter trained on 12,501 AI2D VQA questions conditioned on clean scene graph context. Reproduces 88.74% VQA Acc.

HF Hugging Face Repo
Dataset Hub 45.0 MB

Vi-EduDiag-50 Benchmark

50 high-res (200 DPI) Vietnamese textbook diagrams from Grades 6 to 12 across 4 STEM subjects with bilingual atomic ground truth.

HF Dataset Hub
1-Click Automatic Checkpoint Download & Reproduction
# 1. Download all pretrained checkpoints from Hugging Face Hub:
python scripts/download_checkpoints.py --model all

# 2. Reproduce official 88.74% VQA Accuracy on AI2D test split:
python eval_vqa.py

# 3. Reproduce official 35.67 CIDEr on AI2D-Captioning:
python eval_caption.py

# 4. Reproduce full ablation study matrix & transfer benchmark:
python eval_ablation.py
python eval_transfer.py

Interactive Diagram & Scene Graph Explorer

Explore individual educational diagrams, interactive force-directed graph topologies (Vis.js), vector bounding box overlays, and bilingual pedagogical captions.

Filter by Diagram Category

Select a diagram category to inspect corresponding scientific figures from the official AI2D test split

491 / 491 Diagrams