<?xml version="1.0" encoding="utf-8" standalone="yes"?><rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom" xmlns:content="http://purl.org/rss/1.0/modules/content/"><channel><title>Vision-Language-Models on Hunter Heidenreich | Senior AI Research Scientist</title><link>https://hunterheidenreich.com/tags/vision-language-models/</link><description>Recent content in Vision-Language-Models on Hunter Heidenreich | Senior AI Research Scientist</description><image><title>Hunter Heidenreich | Senior AI Research Scientist</title><url>https://hunterheidenreich.com/img/avatar.webp</url><link>https://hunterheidenreich.com/img/avatar.webp</link></image><generator>Hugo -- 0.163.3</generator><language>en-US</language><copyright>2026 Hunter Heidenreich</copyright><lastBuildDate>Sat, 01 Aug 2026 00:00:00 +0000</lastBuildDate><atom:link href="https://hunterheidenreich.com/tags/vision-language-models/index.xml" rel="self" type="application/rss+xml"/><item><title>MaCBench: Multimodal Chemistry and Materials Benchmark</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/macbench-multimodal-chemistry-benchmark/</link><pubDate>Thu, 26 Mar 2026 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/macbench-multimodal-chemistry-benchmark/</guid><description>MaCBench benchmarks vision language models on chemistry and materials science tasks, revealing failures in spatial reasoning and cross-modal integration.</description><content:encoded><![CDATA[<h2 id="a-benchmark-for-multimodal-scientific-reasoning">A Benchmark for Multimodal Scientific Reasoning</h2>
<p>MaCBench is a <strong>Resource</strong> contribution that provides a comprehensive benchmark for evaluating vision language models (VLLMs) on real-world chemistry and materials science tasks. Rather than testing general-purpose visual reasoning or text-only scientific knowledge, MaCBench specifically targets the interplay between visual and textual modalities across the scientific workflow. The benchmark contains 779 multiple-choice questions and 374 numeric-answer questions organized into 11 topics across three pillars: data extraction, experimental execution, and data interpretation. Through systematic ablation studies, the authors identify fundamental limitations in spatial reasoning, cross-modal synthesis, and multi-step inference that current VLLMs exhibit.</p>
<h2 id="why-multimodal-evaluation-matters-for-chemistry">Why Multimodal Evaluation Matters for Chemistry</h2>
<p>Scientific research inherently requires integrating multiple information modalities: reading plots, interpreting spectra, evaluating laboratory setups, and connecting visual observations with domain knowledge. While text-only benchmarks like <a href="/notes/chemistry/llm-applications/chembench-llm-chemistry-evaluation/">ChemBench</a> have evaluated LLM capabilities in chemistry, and general multimodal benchmarks have tested visual reasoning, no prior work had systematically assessed how VLLMs handle the specific multimodal demands of the chemistry and materials science workflow.</p>
<p>Existing evaluations treated either the scientific reasoning dimension or the multimodal dimension in isolation. This left a critical gap: can VLLMs reliably assist with tasks that require both visual perception and scientific reasoning simultaneously? For example, identifying laboratory equipment is a perception task, but evaluating whether a laboratory setup is safe requires integrating visual understanding with domain-specific knowledge about hazards.</p>
<p>The authors designed MaCBench to fill this gap by constructing tasks that mirror actual scientific workflows and by including ablation studies that isolate specific failure modes.</p>
<h2 id="benchmark-design-three-pillars-of-scientific-work">Benchmark Design: Three Pillars of Scientific Work</h2>
<p>The benchmark is structured around three pillars reflecting the scientific process:</p>
<p><strong>Data Extraction</strong> covers parsing scientific literature, including extracting values from tables and plots, interpreting chemical structure diagrams, and identifying reaction components. Tasks range from simple value extraction to complex spatial reasoning about molecular relationships (e.g., identifying isomeric relationships between compounds).</p>
<p><strong>Experimental Execution</strong> evaluates understanding of laboratory operations and crystallographic analysis. This includes equipment identification, safety assessment of laboratory setups, and interpretation of crystal structure renderings (<a href="https://en.wikipedia.org/wiki/Space_group">space group</a> assignment, atomic species counting, density calculations).</p>
<p><strong>Data Interpretation</strong> tests analysis of experimental outputs: spectral analysis (<a href="https://en.wikipedia.org/wiki/X-ray_diffraction">XRD</a>, <a href="https://en.wikipedia.org/wiki/Nuclear_magnetic_resonance_spectroscopy">NMR</a>, <a href="https://en.wikipedia.org/wiki/Mass_spectrometry">mass spectrometry</a>), electronic structure interpretation, adsorption isotherm analysis, and <a href="https://en.wikipedia.org/wiki/Atomic_force_microscopy">AFM</a> image interpretation.</p>
<p>Each task uses a single prompt template containing multiple questions. All questions pair images with text-based prompts. The dataset was curated manually, with questions reviewed by multiple scientists before inclusion. A BigBench canary string is embedded in each file to prevent data contamination during future model training.</p>
<h2 id="evaluation-of-frontier-vllms-and-ablation-studies">Evaluation of Frontier VLLMs and Ablation Studies</h2>
<p>The authors evaluated four frontier VLLMs: Claude 3.5 Sonnet, GPT-4o, Gemini 1.5 Pro, and Llama 3.2 90B Vision. Performance is reported relative to random baselines to account for the varying number of answer choices across MCQ tasks:</p>
<p>$$
\text{acc}_{\text{rel}} = \text{acc} - \text{acc}_{\text{baseline}}
$$</p>
<p>Each benchmark run was repeated five times to capture variability, with standard deviations reported as error bars.</p>
<h3 id="overall-performance-landscape">Overall Performance Landscape</h3>
<p>Claude 3.5 Sonnet was the leading model across all three task families, though no model dominated across all individual tasks. Key findings:</p>
<ul>
<li><strong>Equipment identification</strong>: average accuracy of 0.77 (strong perception performance)</li>
<li><strong>Hand-drawn molecule to <a href="/notes/chemistry/molecular-representations/notations/smiles/">SMILES</a> matching</strong>: average accuracy of 0.80</li>
<li><strong>Table composition extraction</strong>: average accuracy of 0.53 (Llama 3.2 indistinguishable from random guessing)</li>
<li><strong>Isomer relationship identification</strong>: average accuracy of 0.24 (barely above the 0.14 baseline)</li>
<li><strong>Laboratory safety assessment</strong>: average accuracy of 0.46</li>
<li><strong>AFM image interpretation</strong>: average accuracy of 0.24</li>
<li><strong>NMR and mass spectrometry analysis</strong>: average accuracy of 0.35</li>
</ul>
<h3 id="ablation-studies-four-dimensions-of-failure">Ablation Studies: Four Dimensions of Failure</h3>
<p>The authors designed ablations isolating four specific dimensions:</p>
<p><strong>1. Modality (Image vs. Text):</strong> When identical information was presented as text instead of images, performance improved consistently across all tasks. For XRD peak identification, models showed a roughly 35% performance increase when peaks were provided as text rather than displayed visually. Even crystal structure volume calculations differed by four percentage points between visual and textual input of unit cell parameters.</p>
<p><strong>2. Multi-Step Reasoning:</strong> Performance degraded consistently as tasks required more reasoning steps. For XRD analysis, identifying the highest peak achieved 0.74 average accuracy, while ranking relative peak intensities dropped to 0.28. Isotherm analysis showed the same pattern: finding the maximum value was easier than ordering multiple values.</p>
<p><strong>3. Scientific Terminology:</strong> Removing domain-specific terminology (e.g., using <a href="https://en.wikipedia.org/wiki/IUPAC_nomenclature_of_organic_chemistry">IUPAC names</a> instead of SMILES notation) improved performance on several tasks, suggesting models are sensitive to specific vocabularies rather than understanding underlying concepts. Gemini 1.5 Pro showed particular sensitivity to exact prompt wording, with large performance variations from minor changes like replacing &ldquo;image&rdquo; with &ldquo;diagram&rdquo; or &ldquo;plot.&rdquo;</p>
<p><strong>4. Guidance:</strong> Adding step-by-step instructions improved performance for most models on spectral analysis and XRD pattern matching, with the notable exception of Claude 3.5 Sonnet, whose performance did not improve with guidance.</p>
<h3 id="internet-frequency-correlation">Internet Frequency Correlation</h3>
<p>The authors measured the correlation between model performance and the number of Google search results for various crystal structures (as a proxy for training data frequency). For all tested cases, structures with correct model responses had higher Internet presence. This effect held even for pure perception tasks like counting atomic species, suggesting models may rely on memorized patterns rather than genuine visual reasoning.</p>
<h2 id="limitations-of-current-vllms-for-scientific-assistance">Limitations of Current VLLMs for Scientific Assistance</h2>
<p>The results reveal three fundamental limitations of current VLLMs:</p>
<p><strong>Spatial reasoning failure:</strong> Models perform well on perception tasks (identifying equipment, matching hand-drawn molecules) but fail when spatial understanding is required (<a href="https://en.wikipedia.org/wiki/Stereochemistry">stereochemistry</a> assignment at 0.24 accuracy, space group identification at 0.45). This limitation undermines one of the most intuitive potential use cases of vision models.</p>
<p><strong>Incomplete cross-modal integration:</strong> The consistent performance gap between text and image presentations of identical information demonstrates that current models have not developed robust strategies for visual information processing. The models process text and images through fundamentally different pathways, with text consistently yielding better results.</p>
<p><strong>Multi-step reasoning brittleness:</strong> The systematic degradation across reasoning steps indicates that chaining logical operations, a core requirement for scientific reasoning, remains a fundamental weakness.</p>
<p>The authors note that compared to text-only benchmarks (e.g., ChemBench), multimodal systems show much higher performance variability across tasks, suggesting greater fragility. They propose that advances in synthetic training data generation (particularly for spatial reasoning) and modality transformation training tasks could help address these limitations. They also acknowledge that future workflows with machine-actionable data formats may reduce the need for some multimodal parsing capabilities.</p>
<p>The benchmark does not encompass the full scope of scientific reasoning, and the evaluated models are not exhaustive of all available architectures. The authors call for continued research across wider task and model sets, along with interpretability studies to distinguish genuine reasoning from pattern matching.</p>
<hr>
<h2 id="reproducibility-details">Reproducibility Details</h2>
<h3 id="data">Data</h3>
<table>
	<thead>
			<tr>
					<th>Purpose</th>
					<th>Dataset</th>
					<th>Size</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td>Evaluation</td>
					<td>MaCBench</td>
					<td>779 MCQs + 374 numeric questions</td>
					<td>11 topics across 3 pillars</td>
			</tr>
			<tr>
					<td>Evaluation</td>
					<td>MaCBench-Ablations</td>
					<td>Subset with ablation variants</td>
					<td>Modality, terminology, guidance, step complexity</td>
			</tr>
	</tbody>
</table>
<p>Both datasets are available on HuggingFace. Questions are stored in extended BigBench format with base-64-encoded images and BigBench canary strings.</p>
<h3 id="algorithms">Algorithms</h3>
<p>The evaluation pipeline builds on the ChemBench framework (v0.3.0). Answer extraction uses regex-based parsing backed by an LLM extractor (Claude 3.5 Sonnet) for fallback cases. Refusal detection combines LLM Guard regex patterns with a fine-tuned DistilRoBERTa model, with up to five retries for refused responses.</p>
<p><strong>Scoring:</strong></p>
<ul>
<li>MCQs: correct if <a href="https://en.wikipedia.org/wiki/Hamming_distance">Hamming loss</a> is zero (exact match)</li>
<li>Numeric: correct if mean absolute error falls within specified tolerance (default 1%, up to 5% for specific tasks)</li>
<li>Random baseline: random option selection for MCQs; mean of all target values in a topic for numeric questions</li>
</ul>
<h3 id="models">Models</h3>
<p>Four frontier VLLMs evaluated:</p>
<ul>
<li>Claude 3.5 Sonnet (Anthropic)</li>
<li>GPT-4o (OpenAI)</li>
<li>Gemini 1.5 Pro (Google)</li>
<li>Llama 3.2 90B Vision (Meta)</li>
</ul>
<p>Default quality/resolution settings were used for each provider.</p>
<h3 id="evaluation">Evaluation</h3>
<table>
	<thead>
			<tr>
					<th>Metric</th>
					<th>Best Model</th>
					<th>Value</th>
					<th>Baseline</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td>Equipment identification</td>
					<td>Average</td>
					<td>0.77</td>
					<td>varies</td>
					<td>Near-ceiling perception</td>
			</tr>
			<tr>
					<td>Hand-drawn molecule matching</td>
					<td>Average</td>
					<td>0.80</td>
					<td>~0.20</td>
					<td>4x above baseline</td>
			</tr>
			<tr>
					<td>Isomer relationship</td>
					<td>Average</td>
					<td>0.24</td>
					<td>0.14</td>
					<td>Near random</td>
			</tr>
			<tr>
					<td>Laboratory safety</td>
					<td>Average</td>
					<td>0.46</td>
					<td>varies</td>
					<td>Below practical utility</td>
			</tr>
			<tr>
					<td>AFM interpretation</td>
					<td>Average</td>
					<td>0.24</td>
					<td>varies</td>
					<td>Near random</td>
			</tr>
			<tr>
					<td>Henry constant comparison</td>
					<td>Average</td>
					<td>0.83</td>
					<td>varies</td>
					<td>Strongest interpretation task</td>
			</tr>
	</tbody>
</table>
<h3 id="hardware">Hardware</h3>
<p>The paper does not specify hardware requirements. All evaluations were run through commercial API endpoints.</p>
<h3 id="artifacts">Artifacts</h3>
<table>
	<thead>
			<tr>
					<th>Artifact</th>
					<th>Type</th>
					<th>License</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><a href="https://github.com/lamalab-org/macbench">MaCBench Repository</a></td>
					<td>Code</td>
					<td>MIT</td>
					<td>Benchmark data and evaluation card</td>
			</tr>
			<tr>
					<td><a href="https://github.com/lamalab-org/chembench">ChemBench Framework</a></td>
					<td>Code</td>
					<td>MIT</td>
					<td>Evaluation pipeline (v0.3.0)</td>
			</tr>
			<tr>
					<td><a href="https://huggingface.co/datasets/kjappelbaum/MaCBench">MaCBench Dataset</a></td>
					<td>Dataset</td>
					<td>CC-BY-4.0</td>
					<td>1,153 questions with images</td>
			</tr>
			<tr>
					<td><a href="https://huggingface.co/datasets/kjappelbaum/MaCBench-Ablations">MaCBench-Ablations</a></td>
					<td>Dataset</td>
					<td>CC-BY-4.0</td>
					<td>Ablation task variants</td>
			</tr>
			<tr>
					<td><a href="https://doi.org/10.5281/zenodo.14935487">ChemBench v0.3.0 (Zenodo)</a></td>
					<td>Code</td>
					<td>MIT</td>
					<td>Archived release</td>
			</tr>
	</tbody>
</table>
<hr>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation:</strong> Alampara, N., Schilling-Wilhelmi, M., Ríos-García, M., Mandal, I., Khetarpal, P., Grover, H. S., Krishnan, N. M. A., &amp; Jablonka, K. M. (2025). Probing the limitations of multimodal language models for chemistry and materials research. <em>Nature Computational Science</em>, 5(10), 952-961. <a href="https://doi.org/10.1038/s43588-025-00836-3">https://doi.org/10.1038/s43588-025-00836-3</a></p>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@article</span>{alampara2025macbench,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span>=<span style="color:#e6db74">{Probing the limitations of multimodal language models for chemistry and materials research}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span>=<span style="color:#e6db74">{Alampara, Nawaf and Schilling-Wilhelmi, Mara and R{\&#39;\i}os-Garc{\&#39;\i}a, Marti{\~n}o and Mandal, Indrajeet and Khetarpal, Pranav and Grover, Hargun Singh and Krishnan, N. M. Anoop and Jablonka, Kevin Maik}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">journal</span>=<span style="color:#e6db74">{Nature Computational Science}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">volume</span>=<span style="color:#e6db74">{5}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">number</span>=<span style="color:#e6db74">{10}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">pages</span>=<span style="color:#e6db74">{952--961}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span>=<span style="color:#e6db74">{2025}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">publisher</span>=<span style="color:#e6db74">{Nature Publishing Group}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">doi</span>=<span style="color:#e6db74">{10.1038/s43588-025-00836-3}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div>]]></content:encoded></item><item><title>GutenOCR: A Grounded Vision-Language Front-End for Documents</title><link>https://hunterheidenreich.com/research/gutenocr-grounded-vision-language-frontend/</link><pubDate>Tue, 20 Jan 2026 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/research/gutenocr-grounded-vision-language-frontend/</guid><description>GutenOCR introduces vision-language models for grounded OCR, offering precise text transcription and geometric grounding via a unified prompt-based interface.</description><content:encoded><![CDATA[<p><a href="https://github.com/Roots-Automation/GutenOCR"><strong>GutenOCR</strong></a> is a family of vision-language models (VLMs) designed to serve as a &ldquo;grounded OCR front-end&rdquo;. Unlike traditional OCR pipelines (which are often brittle) or modern &ldquo;OCR-free&rdquo; VLMs (which often lack precise token-to-pixel alignment), GutenOCR is fine-tuned to provide both high-quality text transcription and explicit geometric grounding (bounding boxes) through a unified, prompt-based interface.</p>
<h3 id="abstract">Abstract</h3>
<p>Traditional OCR pipelines are often brittle, while modern &ldquo;OCR-free&rdquo; Vision-Language Models (VLMs) frequently lack precise token-to-pixel alignment. To address this, we introduce <strong>GutenOCR</strong>, a family of VLMs designed specifically as a &ldquo;grounded OCR front-end.&rdquo; By fine-tuning Qwen2.5-VL on a curriculum of synthetic and real-world documents, GutenOCR provides both high-quality text transcription and explicit geometric grounding (bounding boxes) through a unified, prompt-based interface. This approach allows downstream systems to request exactly the data format they need, from plain text to complex JSON structures.</p>
<h3 id="key-contributions--results">Key Contributions &amp; Results</h3>
<ul>
<li><strong>Unified Interface:</strong> Transforms Qwen2.5-VL models into specialized OCR systems supporting full-page reading, detection, localized reading, and conditional detection via prompting.</li>
<li><strong>In-Domain Improvements:</strong> GutenOCR-7B more than doubles the composite grounded OCR score of its base model (0.40 to 0.82) on 10.5K held-out pages, with the gains concentrated in localized reading and detection.</li>
<li><strong>Fox Benchmark:</strong> GutenOCR-3B achieves a region-level Character Error Rate (CER) of 0.053, surpassing even the dedicated Fox model (0.059), and sharply improves line-level OCR over its Qwen2.5-VL backbones (0.817 to 0.240), though the specialized Fox model still leads at line level.</li>
<li><strong>Curriculum Learning:</strong> Training uses a three-stage curriculum across synthetic data, real-world business documents, and long-context scientific articles to progressively build layout and grounding competency.</li>
<li><strong>Trade-offs:</strong> While GutenOCR reads content accurately (high Page F1), it orders text based on 2D layout columns. It also experiences catastrophic forgetting of color-based prompts and clear negative transfer on math formula recognition (notably the 3B model), to the point the paper recommends the un-fine-tuned backbone for equation-heavy use.</li>
</ul>
<h3 id="methodology">Methodology</h3>
<ul>
<li><strong>Data:</strong> The training mixture combines large-scale real-world documents (business forms, scientific articles) with synthetic data designed to teach precise grounding (e.g., &ldquo;Grounded LaTeX&rdquo; and &ldquo;SynthDoG Grounding&rdquo;).</li>
<li><strong>Curriculum Learning:</strong> Training progresses through three stages, starting with short contexts and synthetic data, moving to real-world business documents, and finishing with long-context scientific articles (up to 16k tokens).</li>
<li><strong>Unified Interface:</strong> The model treats &ldquo;pipeline&rdquo; stages (detection, reading, grounding) as different input-output schemas of a single model, allowing downstream systems to request exactly the data format they need (e.g., plain text vs. JSON boxes).</li>
</ul>
<h3 id="models">Models</h3>
<p>We release 3B and 7B parameter models on HuggingFace:</p>
<ul>
<li><a href="https://huggingface.co/bevaya/GutenOCR-3B"><strong>GutenOCR-3B (HuggingFace)</strong></a></li>
<li><a href="https://huggingface.co/bevaya/GutenOCR-7B"><strong>GutenOCR-7B (HuggingFace)</strong></a></li>
</ul>
<p>You can try GutenOCR directly at <a href="https://ocr.bevaya.ai/">ocr.bevaya.ai</a>, where you can upload a document image and see the model&rsquo;s parsed text output alongside bounding-box highlights on the original image.</p>















<figure class="post-figure center ">
    <img src="/img/gutenocr-demo.webp"
         alt="GutenOCR demo interface showing parsed text with hover-to-highlight on the uploaded image"
         title="GutenOCR demo interface showing parsed text with hover-to-highlight on the uploaded image"
         
         
         loading="lazy"
         class="post-image">
    
    <figcaption class="post-caption">The live demo at ocr.bevaya.ai: hovering over any parsed token highlights its bounding box on the original document.</figcaption>
    
</figure>

<h3 id="why-this-matters">Why This Matters</h3>
<p>GutenOCR is proposed as a foundational layer for systems where every extracted answer must be explicitly linked to supporting pixels. By providing stable, grounded outputs, it enables human-in-the-loop workflows where reviewers can easily verify hallucinations or missing text by checking the predicted bounding boxes. This work pairs closely with our release of <a href="/research/pubmed-ocr-pmc-open-access-ocr-annotations/">PubMed-OCR</a>, which provides the large-scale, high-density annotations necessary to train such layout-aware models.</p>
<h3 id="resources">Resources</h3>
<ul>
<li><a href="https://ocr.bevaya.ai/"><strong>Live Demo</strong></a>: Try GutenOCR on your own documents.</li>
<li><a href="https://arxiv.org/abs/2601.14490"><strong>Paper (arXiv)</strong></a>: Full technical report.</li>
<li><a href="https://github.com/Roots-Automation/GutenOCR"><strong>Code (GitHub)</strong></a>: Training code and model release.</li>
<li><a href="https://huggingface.co/bevaya/GutenOCR-3B"><strong>GutenOCR-3B (HuggingFace)</strong></a>: 3B parameter model weights.</li>
<li><a href="https://huggingface.co/bevaya/GutenOCR-7B"><strong>GutenOCR-7B (HuggingFace)</strong></a>: 7B parameter model weights.</li>
</ul>
<h3 id="citation">Citation</h3>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@misc</span>{heidenreich2026gutenocrgroundedvisionlanguagefrontend,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">title</span>=<span style="color:#e6db74">{GutenOCR: A Grounded Vision-Language Front-End for Documents}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">author</span>=<span style="color:#e6db74">{Hunter Heidenreich and Ben Elliott and Olivia Dinica and Yosheb Getachew}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">year</span>=<span style="color:#e6db74">{2026}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">eprint</span>=<span style="color:#e6db74">{2601.14490}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">archivePrefix</span>=<span style="color:#e6db74">{arXiv}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">primaryClass</span>=<span style="color:#e6db74">{cs.CV}</span>,
</span></span><span style="display:flex;"><span>      <span style="color:#a6e22e">url</span>=<span style="color:#e6db74">{https://arxiv.org/abs/2601.14490}</span>,
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div><h3 id="related-work">Related Work</h3>
<ul>
<li><a href="/research/pubmed-ocr-pmc-open-access-ocr-annotations/"><strong>PubMed-OCR</strong></a>: The large-scale annotation dataset used to train GutenOCR&rsquo;s layout-aware grounding capabilities.</li>
<li><a href="/research/llm-page-stream-segmentation/"><strong>LLMs for Page Stream Segmentation</strong></a>: Complementary work on document understanding at the page-stream level.</li>
<li><a href="/posts/history-of-page-stream-segmentation/"><strong>The Evolution of Page Stream Segmentation: Rules to LLMs</strong></a>: Background on the history and evolution of document processing pipelines.</li>
<li><a href="/posts/reliability-trap-document-automation/"><strong>The Reliability Trap: When 99% Accuracy Isn&rsquo;t Enough</strong></a>: Explores calibration challenges in deployed PSS systems, directly relevant to GutenOCR&rsquo;s deployment context as an OCR front-end.</li>
</ul>
]]></content:encoded></item><item><title>Multimodal Search in Chemical Documents and Reactions</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/shah-multimodal-search-2025/</link><pubDate>Sat, 20 Dec 2025 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/shah-multimodal-search-2025/</guid><description>A multimodal search engine that integrates text passages, molecular diagrams, and reaction data to enable passage-level retrieval in chemical literature.</description><content:encoded><![CDATA[<h2 id="contribution-multimodal-synthesis-retrieval">Contribution: Multimodal Synthesis Retrieval</h2>
<p>This paper represents a $\Psi_{\text{Method}}$ projection that proposes a novel architectural pipeline for indexing and searching chemical literature. The framework unifies text, molecular diagrams, and structured reaction records. It also contains a secondary $\Psi_{\text{Resource}}$ projection, providing a functional demonstration tool and curating a specific benchmark dataset for Suzuki coupling reactions.</p>
<h2 id="the-gap-in-passage-level-chemical-retrieval">The Gap in Passage-Level Chemical Retrieval</h2>
<p>Scientific literature documents chemical reactions through a combination of text and visual diagrams. Textual descriptions detail parameters like yield and operational temperature, whereas diagrams graphically model these structural transformations. Existing tools such as SciFinder or <a href="https://en.wikipedia.org/wiki/Reaxys">Reaxys</a> perform document-level or individual compound retrieval. They fail to explicitly link molecular figures to localized textual descriptions. This structure prevents researchers from directly extracting a corresponding reaction diagram alongside the exact textual protocol. Researchers require passage-level retrieval of synthesis protocols to efficiently access complete reaction conditions.</p>
<h2 id="core-innovation-unified-multimodal-indexing">Core Innovation: Unified Multimodal Indexing</h2>
<p>The core methodological innovation is a multimodal passage-level indexing and linking pipeline.</p>
<ul>
<li><strong>Unified Indexing:</strong> The framework processes text and diagrams in parallel and directly links them into a single index structure. This architecture supports search queries utilizing raw text, discrete <a href="/notes/chemistry/molecular-representations/notations/smiles/">SMILES</a> strings, or multimodal combinations.</li>
<li><strong>Compound-Passage Linking:</strong> The mechanism applies conflict-resolution logic linking chemical diagrams to specific text citations using two parallel heuristics:
<ol>
<li><strong>Token-based Alignment:</strong> Matching parsed diagram labels against documented text strings (e.g., &ldquo;compound 5&rdquo;) using normalized <a href="https://en.wikipedia.org/wiki/Levenshtein_distance">Levenshtein distance</a>.</li>
<li><strong>Fingerprint-based Alignment:</strong> Matching chemical structures against generated SMILES strings via structural <a href="https://en.wikipedia.org/wiki/Jaccard_index">Tanimoto Similarity</a>.</li>
</ol>
</li>
<li><strong>ReactionMiner Integration:</strong> The pipeline parses and incorporates formatted reaction records (reactants, products, catalysts, quantitative yields) directly derived from segmented text passages.</li>
</ul>
<h2 id="methodology--expert-evaluation">Methodology &amp; Expert Evaluation</h2>
<p>The authors evaluated the system utilizing a chemical case study targeting specific synthesis domains alongside qualitative expert assessment.</p>
<ul>
<li><strong>Dataset:</strong> Evaluators processed a corpus of 7 research manuscripts and 6 supplementary data documents detailing <a href="https://en.wikipedia.org/wiki/Suzuki_reaction">Suzuki coupling</a> reactions.</li>
<li><strong>Volume:</strong> The resulting index processed 1,282 extracted passages (indexing 538), extracted 383 unique SMILES, and logged 219 parsed reactions.</li>
<li><strong>Qualitative Evaluation:</strong> Practicing structural chemists developed real-world queries (such as cross-referencing the conceptual &ldquo;Burke group&rdquo; alongside an explicit structural SMARTS pattern) to gauge retrieval capability.</li>
</ul>
<h2 id="key-findings--system-limitations">Key Findings &amp; System Limitations</h2>
<ul>
<li><strong>Diagram-to-Text Linking:</strong> The pipeline accurately paired visual molecular diagrams with structurally derived text details, permitting testers to navigate directly from a molecule query card to the exact origin passage within the source PDF.</li>
<li><strong>Contextual Insight Extraction:</strong> Specialized chemists found the parsed reaction representations (yield metrics, isolated catalysts) functionally pragmatic as high-level extractive summaries.</li>
<li><strong>Extrapolative Retrieval:</strong> The architecture permitted the effective retrieval of targeted chemical derivatives (such as benzo[b]thiophen-2-ylboronic acid) via structurally related input queries (dibenzothiophene).</li>
</ul>
<p>The system evaluation highlights several architectural restrictions:</p>
<ul>
<li><strong>Domain-Restricted Validation:</strong> The initial validation is entirely qualitative and bounded to the specific subclass of Suzuki coupling reactions. The evaluation omits standardized quantitative retrieval baselines (e.g., MAP, NDCG) and lacks systematic ablation data for the fusion scoring mechanism.</li>
<li><strong>Algorithmic Transparency:</strong> The multimodal query routing mechanism does not clearly indicate the dominant retrieval feature. This hides whether keyword text or structural similarity actually drove the final result placement. This ambiguity limits operator control.</li>
<li><strong>Optical Processing Brittleness:</strong> The embedded vision inference and primitive parsing pipelines display inherent fragility, producing intermittent failures when associating text passages with correctly parsed molecular diagrams.</li>
<li><strong>Metadata Logging Incompleteness:</strong> Practicing chemists requested additional structured metadata targets (such as specific molar equivalents and parameterized mol% values) to successfully bridge the extracted data stream directly into digital electronic lab notebooks.</li>
</ul>
<hr>
<h2 id="reproducibility">Reproducibility</h2>
<h3 id="artifacts">Artifacts</h3>
<table>
	<thead>
			<tr>
					<th>Artifact</th>
					<th>Type</th>
					<th>License</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><a href="https://www.cs.rit.edu/~dprl/reactionminer-demo-landing/">ReactionMiner Demo</a></td>
					<td>Other</td>
					<td>Unknown</td>
					<td>Online demo landing page; source code repository not publicly linked</td>
			</tr>
	</tbody>
</table>
<h3 id="data">Data</h3>
<ul>
<li><strong>Source:</strong> The corpus features 7 primary research papers and 6 auxiliary supplementary information documents focusing on Suzuki coupling reactions, sourced from practicing chemists at UIUC. This evaluation dataset is strictly internal and not publicly available.</li>
<li><strong>Preprocessing:</strong>
<ul>
<li>Engineers convert source PDFs to full-page raster images.</li>
<li>The system extracts localized graphical layout and raw text via <strong>PyTesseract</strong>.</li>
<li>The pipeline segments valid passage chunks emphasizing reaction-related sentences utilizing product-indicative lexicons and topic modeling.</li>
</ul>
</li>
</ul>
<h3 id="algorithms">Algorithms</h3>
<ul>
<li><strong>Diagram Extraction:</strong> A <strong>YOLOv8</strong> model identifies and segments molecular regions within structured PDF pages.</li>
<li><strong>Diagram Parsing:</strong> The architecture relies on <strong>ChemScraper</strong> to infer structural semantics from raw diagrams:
<ul>
<li><em>Born-digital PDFs:</em> <strong>SymbolScraper</strong> extracts vector lines and polygons directly from bounding box definitions.</li>
<li><em>Raster images:</em> The system employs the <strong>Line Segment Detector (LSD)</strong> and watershed bounding algorithms to isolate native geometric primitives.</li>
</ul>
</li>
<li><strong>Text Entity Extraction:</strong> The framework deploys <strong>ChemDataExtractor 2.0</strong> to extract explicit molecular aliases. A translation layer maps these entities to string representations via <strong>OPSIN</strong>.</li>
<li><strong>Linking Logic (Fusion Score):</strong>
<ul>
<li><strong>Text Link:</strong> The algorithm calculates a normalized Levenshtein ratio connecting visual diagram labels against proximal text mentions based on calculated edit distance.</li>
<li><strong>Structure Link:</strong> The algorithm computes the discrete Tanimoto Similarity between generated 2048-bit Morgan fingerprints extracted from localized visual diagram features and baseline text SMILES queries:
$$ T(A, B) = \frac{A \cdot B}{|A|^{2} + |B|^{2} - A \cdot B} $$
where $A$ and $B$ represent the boolean bit vectors of the respective fingerprint pairs.</li>
<li><strong>Conflict Resolution Protocol:</strong> The system fuses structural geometry bounds and discrete textual tokenization metrics, prioritizing the ranking sequence that yields a higher terminal similarity score. During final retrieval, the candidate subset is systematically re-ranked leveraging the hybrid calculation of the <a href="https://en.wikipedia.org/wiki/Okapi_BM25">BM25</a> explicit metric and the localized count of exact SMILES pattern hits.</li>
</ul>
</li>
</ul>
<h3 id="models">Models</h3>
<ul>
<li><strong>Reaction Extraction Parameters:</strong> The engineers configure a <strong>LLaMA-3.1-8b</strong> model fine-tuned entirely via <strong>LoRA</strong> targeting custom tokens representing reaction entities (compounds, reagents, thermal inputs) directly pulled from text sub-chunks. Exact prompt constraints, the fine-tuning dataset, and specific LoRA hyperparameters are omitted from the source text.</li>
<li><strong>Diagram Processing Bounds:</strong> The codebase incorporates a segmentation-aware multi-task neural network topology built into ChemScraper to execute low-level raster image parsing tasks.</li>
</ul>
<h3 id="evaluation">Evaluation</h3>
<ul>
<li><strong>Search Engine Base:</strong> The authors implemented their indexing framework scaling atop <strong>PyTerrier</strong>.</li>
<li><strong>Text Feature Ranking:</strong> The metric utilizes standalone <strong>BM25</strong> bounds mapping keyword-similarity.</li>
<li><strong>Structure Feature Operations:</strong> The topology operates <strong>RDKit</strong> bindings powering substructure coordinate mapping logic and exact molecular similarity searches.</li>
<li><strong>Multimodal Fusion Processing:</strong>
<ul>
<li>The algorithm filters out terminal candidates mapping initial structural properties (SMILES queries) against the document-wide lexical properties (BM25 scores).</li>
<li>The final fusion routing assigns the strongest positive weight to retrieved passages that accumulate dense local clusters of structurally exact verified SMILES patterns.</li>
</ul>
</li>
</ul>
<h3 id="hardware">Hardware</h3>
<ul>
<li><strong>Compute Infrastructure:</strong> The hardware and parameter requirements to host the multi-stage vision extractors (YOLOv8, ChemScraper) alongside a local 8B LLM are entirely unspecified in the paper.</li>
</ul>
<hr>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation</strong>: Shah, A. K., et al. (2025). Multimodal Search in Chemical Documents and Reactions. In <em>Proceedings of the 48th International ACM SIGIR Conference on Research and Development in Information Retrieval (SIGIR &lsquo;25)</em>. ACM. <a href="https://doi.org/10.48550/arXiv.2502.16865">https://doi.org/10.48550/arXiv.2502.16865</a></p>
<p><strong>Publication</strong>: SIGIR &lsquo;25 (Demo Track), 2025</p>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@misc</span>{shahMultimodalSearchChemical2025,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span> = <span style="color:#e6db74">{Multimodal {{Search}} in {{Chemical Documents}} and {{Reactions}}}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span> = <span style="color:#e6db74">{Shah, Ayush Kumar and Dey, Abhisek and Luo, Leo and Amador, Bryan and Philippy, Patrick and Zhong, Ming and Ouyang, Siru and Friday, David Mark and Bianchi, David and Jackson, Nick and Zanibbi, Richard and Han, Jiawei}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span> = <span style="color:#ae81ff">2025</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">month</span> = feb,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">number</span> = <span style="color:#e6db74">{arXiv:2502.16865}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">eprint</span> = <span style="color:#e6db74">{2502.16865}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">primaryclass</span> = <span style="color:#e6db74">{cs}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">publisher</span> = <span style="color:#e6db74">{arXiv}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">doi</span> = <span style="color:#e6db74">{10.48550/arXiv.2502.16865}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">archiveprefix</span> = <span style="color:#e6db74">{arXiv}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div><p><strong>Additional Resources</strong>:</p>
<ul>
<li><a href="https://www.cs.rit.edu/~dprl/reactionminer-demo-landing/">Online Demo</a> (Note: While the landing page advertises the system as open-source, the exact repository URL and installation prerequisites are omitted from the official manuscript.)</li>
</ul>
]]></content:encoded></item><item><title>MERMaid: Multimodal Chemical Reaction Mining from PDFs</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/mermaid/</link><pubDate>Sat, 20 Dec 2025 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/mermaid/</guid><description>Vision-language pipeline extracting chemical reaction data from PDF figures and tables into structured knowledge graphs with 87% accuracy.</description><content:encoded><![CDATA[<h2 id="methodological-and-resource-contributions">Methodological and Resource Contributions</h2>
<p>This is primarily a <strong>Methodological</strong> paper ($\Psi_{\text{Method}}$) that introduces a novel pipeline (MERMaid) for extracting structured chemical data from unstructured PDF documents. It proposes a specific architecture combining fine-tuned vision models (VisualHeist) with vision-language models (DataRaider) and a retrieval-augmented generation system (KGWizard) to solve the problem of multimodal data ingestion.</p>
<p>Secondarily, it is a <strong>Resource</strong> paper ($\Psi_{\text{Resource}}$) as it releases the source code, prompts, and a new benchmark dataset (<strong>MERMaid-100</strong>) consisting of annotated reaction data across three chemical domains.</p>
<h2 id="the-inaccessibility-of-diagrammatic-reaction-data">The Inaccessibility of Diagrammatic Reaction Data</h2>
<ul>
<li><strong>Data Inaccessibility</strong>: A significant volume of chemical knowledge currently resides in &ldquo;print-optimized&rdquo; PDF formats, specifically within graphical elements like figures, schemes, and tables, which resist standard text mining.</li>
<li><strong>Limitations of Prior Work</strong>: Existing tools (e.g., ChemDataExtractor, <a href="/notes/chemistry/optical-structure-recognition/image-to-graph/molmole/">OpenChemIE</a>) focus primarily on text, struggle with multimodal parsing, or lack the &ldquo;contextual awareness&rdquo; needed to interpret implicit information (e.g., &ldquo;standard conditions&rdquo; with modifications in optimization tables).</li>
<li><strong>Need for Structured Data</strong>: To enable <a href="/notes/chemistry/llm-applications/autonomous-chemical-research-coscientist/">self-driving laboratories</a> and data-driven discovery, this unstructured literature must be converted into machine-actionable formats like <a href="https://en.wikipedia.org/wiki/Knowledge_graph">knowledge graphs</a>.</li>
</ul>
<h2 id="the-mermaid-pipeline-vision-models-and-llm-rag">The MERMaid Pipeline: Vision Models and LLM RAG</h2>
<ul>
<li><strong>VisualHeist (Fine-tuned Segmentation)</strong>: A custom fine-tuned model based on Microsoft&rsquo;s Florence-2 that accurately segments figures, captions, and footnotes, even in messy supplementary materials.</li>
<li><strong>DataRaider (Context-Aware Extraction)</strong>: A VLM-powered module (using GPT-4o) with a <strong>two-step prompt framework</strong> that performs &ldquo;self-directed context completion.&rdquo; It can infer missing reaction parameters from context and resolve footnote labels (e.g., linking &ldquo;condition a&rdquo; in a table to its footnote description).</li>
<li><strong>KGWizard (Schema-Adaptive Graph Construction)</strong>: A text-to-graph engine that uses LLMs as higher-order functions to synthesize parsers dynamically. It employs <strong>Retrieval-Augmented Generation (RAG)</strong> to check for existing nodes during creation, implicitly resolving coreferences (e.g., unifying &ldquo;MeCN&rdquo; and &ldquo;Acetonitrile&rdquo;).</li>
<li><strong>Topic-Agnostic Design</strong>: MERMaid features a flexible design that works across three distinct domains: <a href="https://en.wikipedia.org/wiki/Electrosynthesis">organic electrosynthesis</a>, <a href="https://en.wikipedia.org/wiki/Photocatalysis">photocatalysis</a>, and organic synthesis.</li>
</ul>
<h2 id="benchmarking-segmentation-and-extraction-accuracy">Benchmarking Segmentation and Extraction Accuracy</h2>
<ul>
<li><strong>Segmentation Benchmarking</strong>: The authors compared VisualHeist against OpenChemIE (LayoutParser) and PDFigCapX using a dataset of 121 PDFs from 5 publishers.</li>
<li><strong>End-to-End Extraction</strong>: Evaluated the full pipeline on <strong>MERMaid-100</strong>, a curated dataset of 100 articles across three domains (organic electrosynthesis, photocatalysis, organic synthesis).
<ul>
<li>Validating extraction of specific parameters (e.g., catalysts, solvents, yields) using &ldquo;hard-match&rdquo; accuracy.</li>
</ul>
</li>
<li><strong>Knowledge Graph Construction</strong>: Automatically generated knowledge graphs for the three domains and assessed the structural integrity and <a href="https://en.wikipedia.org/wiki/Coreference">coreference resolution</a> accuracy.</li>
</ul>
<h2 id="end-to-end-extraction-performance">End-to-End Extraction Performance</h2>
<ul>
<li><strong>Segmentation Results</strong>: VisualHeist achieved &gt;93% F1 score across all document types (including pre-2000 papers and supplementary materials), outperforming OpenChemIE by 15-75% and PDFigCapX by 28-75% across all metrics.</li>
<li><strong>Extraction Accuracy</strong>: DataRaider achieved &gt;92% accuracy for VLM-based parameter extraction and near-unity accuracy for domain-specific reaction parameters (e.g., anode, cathode, photocatalyst).</li>
<li><strong>Graph Building</strong>: KGWizard achieved 96% accuracy in node creation and coreference resolution.</li>
<li><strong>Overall Performance</strong>: The pipeline demonstrated an 87% end-to-end overall accuracy.</li>
<li><strong>Limitations</strong>: The architecture relies heavily on closed-weight models (GPT-4o) for reasoning and graph construction, which risks future reproducibility if API snapshots are deprecated. Additionally, the system remains vulnerable to cumulative error propagation from upstream OCR/OCSR tools like <a href="/notes/chemistry/optical-structure-recognition/benchmarks/ocsr-methods/">RxnScribe</a>.</li>
<li><strong>Availability</strong>: The authors provide a modular, extensible framework that can be adapted to other scientific domains.</li>
</ul>
<hr>
<h2 id="reproducibility-details">Reproducibility Details</h2>
<h3 id="data">Data</h3>
<ul>
<li><strong>Training Data (VisualHeist)</strong>:
<ul>
<li>Dataset of <strong>3,435 figures</strong> and <strong>1,716 tables</strong> annotated from 3,518 PDF pages.</li>
<li>Includes main text, supplementary materials, and unformatted archive papers.</li>
</ul>
</li>
<li><strong>Evaluation Data (MERMaid-100)</strong>:
<ul>
<li><strong>100 PDF articles</strong> curated from three domains: organic electrosynthesis, photocatalysis, and organic synthesis.</li>
<li>Includes 104 image-caption/table-heading pairs relevant to reaction optimization.</li>
<li>Available for download at Zenodo (DOI: 10.5281/zenodo.14917752).</li>
</ul>
</li>
</ul>
<h3 id="algorithms">Algorithms</h3>
<ul>
<li><strong>Two-Step Prompt Framework (DataRaider)</strong>:
<ul>
<li><em>Step 1</em>: Generic base prompt + domain keys to extract &ldquo;reaction dictionaries&rdquo; and &ldquo;footnote dictionaries&rdquo;. Uses &ldquo;fill-in-the-blank&rdquo; inference for missing details.</li>
<li><em>Step 2</em>: Safety check prompt where the VLM updates the reaction dictionary using the footnote dictionary to resolve entry-specific modifications.</li>
</ul>
</li>
<li><strong>LLM-Synthesized Parsers (KGWizard)</strong>:
<ul>
<li>Uses LLM as a function $g_{A,B}: A \times B \rightarrow (X \rightarrow Y)$ to generate Python code (parsers) dynamically based on input schema instructions.</li>
</ul>
</li>
<li><strong>RAG for Coreference</strong>:
<ul>
<li>During graph construction, the system queries the existing database for matching values (e.g., &ldquo;MeCN&rdquo;) before creating new nodes to prevent duplication.</li>
</ul>
</li>
<li><strong>Batching</strong>:
<ul>
<li>Articles processed in dynamic batch sizes (starting at 1, increasing to 30) to balance speed and redundancy checks.</li>
</ul>
</li>
</ul>
<h3 id="models">Models</h3>
<ul>
<li><strong>VisualHeist</strong>: Fine-tuned <strong>Florence-2-large</strong> (Microsoft vision foundation model).
<ul>
<li><em>Hyperparameters</em>: 12 epochs, learning rate $5 \times 10^{-6}$, batch size 4.</li>
</ul>
</li>
<li><strong>DataRaider &amp; KGWizard</strong>: <strong>GPT-4o</strong> (version <code>gpt-4o-2024-08-06</code>). Note: Requires an active OpenAI API key. The pipeline&rsquo;s long-term reproducibility is currently tied to the continued availability of this specific closed-source endpoint.</li>
<li><strong>RxnScribe</strong>: Used for <a href="/notes/chemistry/optical-structure-recognition/benchmarks/ocsr-methods/">Optical Chemical Structure Recognition (OCSR)</a> to convert reactant/product images to <a href="/notes/chemistry/molecular-representations/notations/smiles/">SMILES</a>.</li>
</ul>
<h3 id="evaluation">Evaluation</h3>
<ul>
<li><strong>Metrics</strong>:
<ul>
<li><em>Segmentation</em>: Precision, Recall, F1, Accuracy.</li>
<li><em>Caption Extraction</em>: Evaluated via <a href="https://en.wikipedia.org/wiki/Jaccard_index">Jaccard similarity</a>, mapping predicted token sets $A$ and true token sets $B$ to a threshold condition: $$J(A, B) = \frac{|A \cap B|}{|A \cup B|} \ge 0.70$$</li>
<li><em>Data Extraction</em>: Evaluated via Hard-Match accuracy, requiring exact correspondence between predicted sets ($\hat{Y}$) and ground-truth parameters ($Y$) for specific roles (e.g., anode vs. cathode): $$\text{HMA} = \frac{1}{|N|} \sum_{i=1}^{N} \mathbb{1}[y_i = \hat{y}_i]$$</li>
</ul>
</li>
<li><strong>Baselines</strong>: OpenChemIE (LayoutParser + EasyOCR) and PDFigCapX.</li>
</ul>
<h3 id="hardware">Hardware</h3>
<ul>
<li><strong>Training (VisualHeist)</strong>: 2x NVLINK Nvidia RTX A6000 GPUs (48GB VRAM) + Intel Xeon w7-2495X CPU (48 cores).</li>
<li><strong>DataRaider Evaluation</strong>: 13th Gen Intel Core i7-1360P CPU (12 cores).</li>
<li><strong>Inference Costs</strong>:
<ul>
<li>DataRaider: ~$0.051 per image.</li>
<li>KGWizard: ~$0.40 per JSON.</li>
</ul>
</li>
<li><strong>Timing</strong>:
<ul>
<li>VisualHeist inference: ~4.5 seconds/image.</li>
<li>DataRaider inference: ~41.3 seconds/image.</li>
<li>KGWizard processing: ~110.6 seconds/file.</li>
</ul>
</li>
</ul>
<hr>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation</strong>: Leong, S. X., Pablo-García, S., Wong, B., &amp; Aspuru-Guzik, A. (2025). MERMaid: Universal multimodal mining of chemical reactions from PDFs using vision-language models. <em>Matter</em>, 8(12), 102331. <a href="https://doi.org/10.1016/j.matt.2025.102331">https://doi.org/10.1016/j.matt.2025.102331</a></p>
<p><strong>Publication</strong>: Matter, 2025</p>
<p><strong>Artifacts</strong>:</p>
<table>
	<thead>
			<tr>
					<th>Artifact</th>
					<th>Type</th>
					<th>License</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><a href="https://github.com/aspuru-guzik-group/MERMaid">GitHub Repository</a></td>
					<td>Code</td>
					<td>MIT</td>
					<td>Official implementation (VisualHeist, DataRaider, KGWizard)</td>
			</tr>
			<tr>
					<td><a href="https://doi.org/10.5281/zenodo.14917752">Zenodo Data/Prompts</a></td>
					<td>Dataset</td>
					<td>Unknown</td>
					<td>MERMaid-100 benchmark, prompts, and raw VLM responses</td>
			</tr>
	</tbody>
</table>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@article</span>{leong2025mermaid,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span>=<span style="color:#e6db74">{MERMaid: Universal multimodal mining of chemical reactions from PDFs using vision-language models}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span>=<span style="color:#e6db74">{Leong, Shi Xuan and Pablo-Garc{\&#39;i}a, Sergio and Wong, Brandon and Aspuru-Guzik, Al{\&#39;a}n}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">journal</span>=<span style="color:#e6db74">{Matter}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">volume</span>=<span style="color:#e6db74">{8}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">number</span>=<span style="color:#e6db74">{12}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">pages</span>=<span style="color:#e6db74">{102331}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span>=<span style="color:#e6db74">{2025}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">doi</span>=<span style="color:#e6db74">{10.1016/j.matt.2025.102331}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div>]]></content:encoded></item><item><title>InstructMol: Multi-Modal Molecular LLM for Drug Discovery</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/instructmol/</link><pubDate>Sat, 20 Dec 2025 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/instructmol/</guid><description>A multi-modal LLM aligning 2D molecular graphs with text via two-stage instruction tuning for drug discovery tasks.</description><content:encoded><![CDATA[<h2 id="instructmol-framework-overview">InstructMol Framework Overview</h2>
<p><strong>Methodological Paper ($\Psi_{\text{Method}}$)</strong></p>
<p>This work proposes <strong>InstructMol</strong>, a novel multi-modal architecture and training paradigm. It focuses on engineering a system that aligns a pre-trained molecular graph encoder with a general-purpose Large Language Model (LLM). The paper&rsquo;s primary contribution is the <strong>Two-Stage Instruction Tuning</strong> strategy (Alignment Pre-training + Task-Specific Tuning) designed to bridge the modality gap between 2D molecular graphs and natural language.</p>
<h2 id="bridging-specialist-and-generalist-models">Bridging Specialist and Generalist Models</h2>
<p>Current AI approaches in drug discovery typically fall into two categories. Specialist models deliver high accuracy on specific tasks (such as property prediction) but require extensive labeled datasets and lack conversational adaptability. Conversely, generalist LLMs offer strong reasoning and dialogue capabilities but struggle to natively interpret complex structural data, often relying on brittle 1D text representations of molecules like <a href="/notes/chemistry/molecular-representations/notations/smiles/">SMILES</a>.</p>
<p>There is a practical need for a unified &ldquo;Molecular Assistant&rdquo; capable of visually interpreting molecular graphs, reasoning about structure in natural language, and adapting across tasks like synthesis planning and property analysis without training from scratch.</p>
<h2 id="two-stage-modality-alignment">Two-Stage Modality Alignment</h2>
<p>The core novelty lies in the architecture and the <strong>two-stage training pipeline</strong> designed to align differing modalities efficiently:</p>
<ol>
<li><strong>MoleculeSTM Integration</strong>: InstructMol initializes its graph encoder with <strong>MoleculeSTM</strong>, which is already pre-aligned with text via contrastive learning, facilitating easier downstream alignment.</li>
<li><strong>Two-Stage Alignment Strategy</strong>:
<ul>
<li><strong>Stage 1 (Alignment Pre-training)</strong>: Freezes both the LLM and Graph Encoder; trains <em>only</em> a linear projector using a massive dataset of molecule-description pairs to map graph features into the LLM&rsquo;s token space.</li>
<li><strong>Stage 2 (Task-Specific Instruction Tuning)</strong>: Freezes the Graph Encoder; fine-tunes the Projector and the LLM (using <strong>LoRA</strong>) on specific downstream tasks. This allows the model to adapt its reasoning capabilities while preserving the structural understanding gained in Stage 1.</li>
</ul>
</li>
</ol>
<h2 id="task-evaluation-in-drug-discovery">Task Evaluation in Drug Discovery</h2>
<p>The authors evaluated InstructMol across three distinct categories of drug discovery tasks, comparing it against generalist LLMs (Vicuna, LLaMA, <a href="/notes/chemistry/llm-applications/galactica-large-language-model-for-science/">Galactica</a>) and specialist models (<a href="/notes/chemistry/molecular-representations/encoders/chemberta/">ChemBERTa</a>, MolT5):</p>
<ol>
<li><strong>Property Prediction</strong>:
<ul>
<li><em>Regression</em>: Predicting quantum mechanical properties (HOMO, LUMO, Gap) using the <a href="/notes/chemistry/datasets/qm9/">QM9</a> dataset.</li>
<li><em>Classification</em>: Predicting biological activity (BACE, BBBP, HIV) using <a href="/notes/chemistry/molecular-design/property-prediction/moleculenet-benchmark-molecular-ml/">MoleculeNet</a>.</li>
</ul>
</li>
<li><strong>Molecule Description Generation</strong>: Generating natural language descriptions of molecules using the ChEBI-20 dataset.</li>
<li><strong>Chemical Reaction Analysis</strong>:
<ul>
<li><em>Forward Reaction Prediction</em>: Predicting products from reactants.</li>
<li><em>Reagent Prediction</em>: Identifying necessary reagents.</li>
<li><em><a href="https://en.wikipedia.org/wiki/Retrosynthetic_analysis">Retrosynthesis</a></em>: Suggesting reactants for a given product.</li>
</ul>
</li>
</ol>
<p><strong>Ablation Studies</strong> tested the impact of the projector type (Linear vs. MLP), LLM scale (7B vs 13B), and the necessity of the two-stage training approach.</p>
<h2 id="core-findings-and-limitations">Core Findings and Limitations</h2>
<ul>
<li><strong>Improvement Over Baseline Generalists</strong>: InstructMol significantly outperformed generalist LLMs (like LLaMA and Galactica) on all tasks, demonstrating the value of incorporating explicit graph modalities.</li>
<li><strong>Reducing the Gap with Specialists</strong>: While InstructMol brings versatile reasoning capabilities, it still trails highly optimized specialist models (such as Uni-Mol and MolT5) on tasks like molecule description generation. This remaining gap likely stems from its reliance on a relatively small alignment pre-training dataset (~264K PubChem pairs) and the information bottleneck of using a simple linear projector, compared to the millions of structures used to train expert foundational models.</li>
<li><strong>Importance of Alignment</strong>: Ablation studies confirmed that skipping Stage 1 (Alignment Pre-training) degraded performance, proving that a dedicated phase for projecting graph features into text space is crucial.</li>
<li><strong>Limitation</strong>: The model struggles with highly imbalanced datasets (e.g., HIV) and complex reaction mixtures where mapping multiple graph tokens to text becomes ambiguous.</li>
</ul>
<hr>
<h2 id="reproducibility-details">Reproducibility Details</h2>
<h3 id="data">Data</h3>
<p>The training pipeline utilizes distinct datasets for the two stages. <strong>Note:</strong> As of the latest repository update, the finely-processed instruction-tuning datasets (e.g., the filtered ~264K PubChem pairs and instruction-formatted subset pairs) are listed as &ldquo;coming soon&rdquo;, requiring manual recreation for full reproduction.</p>
<table>
	<thead>
			<tr>
					<th style="text-align: left">Purpose</th>
					<th style="text-align: left">Dataset</th>
					<th style="text-align: left">Size</th>
					<th style="text-align: left">Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td style="text-align: left"><strong>Stage 1</strong> (Alignment)</td>
					<td style="text-align: left"><strong><a href="https://en.wikipedia.org/wiki/PubChem">PubChem</a></strong></td>
					<td style="text-align: left">~264K pairs</td>
					<td style="text-align: left">Molecule-text pairs. Filtered from 330K for invalid descriptions and overlaps with ChEBI-20 test set.</td>
			</tr>
			<tr>
					<td style="text-align: left"><strong>Stage 2</strong> (Prop. Reg.)</td>
					<td style="text-align: left"><strong>QM9</strong></td>
					<td style="text-align: left">362K samples</td>
					<td style="text-align: left">Quantum mechanics properties (HOMO, LUMO, Gap).</td>
			</tr>
			<tr>
					<td style="text-align: left"><strong>Stage 2</strong> (Prop. Class.)</td>
					<td style="text-align: left"><strong>MoleculeNet</strong></td>
					<td style="text-align: left">35K samples</td>
					<td style="text-align: left">BACE, BBBP, HIV datasets. Converted to instruction format (Yes/No answer).</td>
			</tr>
			<tr>
					<td style="text-align: left"><strong>Stage 2</strong> (Generation)</td>
					<td style="text-align: left"><strong>ChEBI-20</strong></td>
					<td style="text-align: left">26.5K samples</td>
					<td style="text-align: left">Molecule description generation.</td>
			</tr>
			<tr>
					<td style="text-align: left"><strong>Stage 2</strong> (Reactions)</td>
					<td style="text-align: left"><strong>USPTO</strong></td>
					<td style="text-align: left">~380K samples</td>
					<td style="text-align: left">Combined datasets for Forward (125K), Retrosynthesis (130K), and Reagent (125K) prediction.</td>
			</tr>
	</tbody>
</table>
<h3 id="algorithms">Algorithms</h3>
<ul>
<li><strong>Two-Stage Training</strong>:
<ol>
<li><strong>Alignment Pre-training</strong>: Updates only the Projector. The objective maximizes the probability of generating the target description token sequence $\mathbf{X}_A$ given the molecule input $\mathbf{X}_M$ and instruction $\mathbf{X}_I$:
$$p(\mathbf{X}_A | \mathbf{X}_M, \mathbf{X}_I) = \prod_{i=1}^L p_\theta(x_i | \mathbf{X}_G \parallel \mathbf{X}_S, \mathbf{X}_I, \mathbf{X}_{A,&lt;i})$$</li>
<li><strong>Instruction Tuning</strong>: Updates Projector + LLM (via LoRA) using standard autoregressive language modeling on task-specific instructions. The objective minimizes the negative log-likelihood of generating the target response $R$ of length $L$:
$$\mathcal{L}(\theta) = -\sum_{i=1}^L \log p(R_i | I, M, R_{&lt;i}; \theta)$$
where $I$ represents the instruction and $M$ is the multi-modal molecular input.</li>
</ol>
</li>
<li><strong>LoRA (Low-Rank Adaptation)</strong>: Applied to the LLM in Stage 2. Rank $r=64$, Scaling $\alpha=16$.</li>
<li><strong>Optimization</strong>: AdamW optimizer. Learning rate starts at 2e-3 (Stage 1) and 8e-5 (Stage 2) with cosine decay. Warm-up ratio 0.03.</li>
</ul>
<h3 id="models">Models</h3>
<p><strong>Note:</strong> The official repository currently lists the final fine-tuned <strong>InstructMol weights</strong> as &ldquo;coming soon.&rdquo; Consequently, one must fine-tune the components using the provided scripts. Base model weights (Vicuna-7B and MoleculeSTM) are publicly available via Hugging Face.</p>
<ul>
<li><strong>Graph Encoder ($f_g$)</strong>:
<ul>
<li>Architecture: Graph Isomorphism Network (GIN) with 5 layers.</li>
<li>Hidden Dimension: 300.</li>
<li>Initialization: <strong>MoleculeSTM</strong> checkpoint (pre-trained via contrastive learning).</li>
<li>Status: <strong>Frozen</strong> during Stage 2.</li>
</ul>
</li>
<li><strong>LLM</strong>:
<ul>
<li>Base: <strong>Vicuna-v1.3-7B</strong>.</li>
<li>Status: Frozen in Stage 1; LoRA fine-tuned in Stage 2.</li>
</ul>
</li>
<li><strong>Projector</strong>:
<ul>
<li>Architecture: Linear Layer.</li>
<li>Function: Maps node-level graph representation $Z_G \in \mathbb{R}^{N \times d}$ to the LLM&rsquo;s word embedding space dimensions.</li>
</ul>
</li>
</ul>
<h3 id="evaluation">Evaluation</h3>
<ul>
<li><strong>Metric Libraries</strong>: RDKit for validity/fingerprints, standard NLP libraries for BLEU/ROUGE.</li>
<li><strong>Reaction Metrics</strong>: Fingerprint <a href="https://en.wikipedia.org/wiki/Jaccard_index">Tanimoto Similarity</a> (FTS), Exact Match, Levenshtein distance, and validity (via RDKit).</li>
<li><strong>Description Metrics</strong>: BLEU-2, BLEU-4, ROUGE-1, ROUGE-2, ROUGE-L, METEOR.</li>
</ul>
<h3 id="hardware">Hardware</h3>
<ul>
<li><strong>Compute</strong>: 4 x NVIDIA RTX A6000 (48GB VRAM).</li>
<li><strong>Training Time</strong>:
<ul>
<li>Stage 1: 5 epochs.</li>
<li>Stage 2: 20-50 epochs (Description Generation), 10 epochs (Properties/Reactions).</li>
</ul>
</li>
<li><strong>Batch Size</strong>: 128 for both stages.</li>
</ul>
<h3 id="artifacts">Artifacts</h3>
<table>
	<thead>
			<tr>
					<th style="text-align: left">Artifact</th>
					<th style="text-align: left">Type</th>
					<th style="text-align: left">License</th>
					<th style="text-align: left">Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td style="text-align: left"><a href="https://github.com/IDEA-XL/InstructMol">InstructMol (GitHub)</a></td>
					<td style="text-align: left">Code</td>
					<td style="text-align: left">Apache 2.0 (code), CC BY-NC 4.0 (data)</td>
					<td style="text-align: left">Training/evaluation scripts provided; fine-tuned weights listed as &ldquo;coming soon&rdquo;</td>
			</tr>
			<tr>
					<td style="text-align: left"><a href="https://huggingface.co/lmsys/vicuna-7b-v1.3">Vicuna-7B v1.3</a></td>
					<td style="text-align: left">Model</td>
					<td style="text-align: left">Non-commercial (LLaMA license)</td>
					<td style="text-align: left">Base LLM; must be downloaded separately</td>
			</tr>
			<tr>
					<td style="text-align: left"><a href="https://huggingface.co/chao1224/MoleculeSTM">MoleculeSTM</a></td>
					<td style="text-align: left">Model</td>
					<td style="text-align: left">MIT</td>
					<td style="text-align: left">Pre-trained graph encoder checkpoint</td>
			</tr>
	</tbody>
</table>
<hr>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation</strong>: Cao, H., Liu, Z., Lu, X., Yao, Y., &amp; Li, Y. (2025). InstructMol: Multi-Modal Integration for Building a Versatile and Reliable Molecular Assistant in Drug Discovery. <em>Proceedings of the 31st International Conference on Computational Linguistics</em>, 354-379.</p>
<p><strong>Publication</strong>: COLING 2025</p>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@inproceedings</span>{caoInstructMolMultiModalIntegration2025,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span> = <span style="color:#e6db74">{{{InstructMol}}: {{Multi-Modal Integration}} for {{Building}} a {{Versatile}} and {{Reliable Molecular Assistant}} in {{Drug Discovery}}}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">shorttitle</span> = <span style="color:#e6db74">{{{InstructMol}}}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">booktitle</span> = <span style="color:#e6db74">{Proceedings of the 31st {{International Conference}} on {{Computational Linguistics}}}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span> = <span style="color:#e6db74">{Cao, He and Liu, Zijing and Lu, Xingyu and Yao, Yuan and Li, Yu}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">editor</span> = <span style="color:#e6db74">{Rambow, Owen and Wanner, Leo and Apidianaki, Marianna and {Al-Khalifa}, Hend and Eugenio, Barbara Di and Schockaert, Steven}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span> = <span style="color:#ae81ff">2025</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">month</span> = jan,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">pages</span> = <span style="color:#e6db74">{354--379}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">url</span> = <span style="color:#e6db74">{https://aclanthology.org/2025.coling-main.25/}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">publisher</span> = <span style="color:#e6db74">{Association for Computational Linguistics}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">address</span> = <span style="color:#e6db74">{Abu Dhabi, UAE}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">abstract</span> = <span style="color:#e6db74">{The rapid evolution of artificial intelligence in drug discovery encounters challenges with generalization and extensive training, yet Large Language Models (LLMs) offer promise in reshaping interactions with complex molecular data. Our novel contribution, InstructMol, a multi-modal LLM, effectively aligns molecular structures with natural language via an instruction-tuning approach, utilizing a two-stage training strategy that adeptly combines limited domain-specific data with molecular and textual information. InstructMol showcases substantial performance improvements in drug discovery-related molecular tasks, surpassing leading LLMs and significantly reducing the gap with specialists, thereby establishing a robust foundation for a versatile and dependable drug discovery assistant.}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div><p><strong>Additional Resources</strong>:</p>
<ul>
<li><a href="https://github.com/IDEA-XL/InstructMol">Official Repository</a></li>
</ul>
]]></content:encoded></item><item><title>ChemDFM-X: Multimodal Foundation Model for Chemistry</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/chemdfm-x/</link><pubDate>Sat, 20 Dec 2025 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/chemdfm-x/</guid><description>Multimodal chemical model integrating 5 modalities (2D graphs, 3D conformations, images, MS2/IR spectra) trained on 7.6M instructions.</description><content:encoded><![CDATA[<h2 id="chemdfm-x-contribution-and-architecture">ChemDFM-X Contribution and Architecture</h2>
<p>This is primarily a <strong>Method</strong> paper with a significant <strong>Resource</strong> contribution.</p>
<p><strong>Method</strong>: The paper proposes a novel &ldquo;Cross-modal Dialogue Foundation Model&rdquo; architecture that aligns five distinct chemical modalities (2D graphs, 3D conformations, images, MS2 spectra, IR spectra) to a single LLM decoder using separate encoders and projection modules. It establishes strong baseline performance across multiple modalities compared against current generalist models.</p>
<p><strong>Resource</strong>: The paper addresses the scarcity of multimodal chemical data by constructing a <strong>7.6M instruction-tuning dataset</strong>. This dataset is largely synthesized from seed SMILES strings using approximate calculations (MMFF94, CFM-ID, Chemprop-IR) and specialist model predictions.</p>
<h2 id="bridging-experimental-data-and-llms">Bridging Experimental Data and LLMs</h2>
<p>Existing chemical AI models generally fall into two distinct categories. Task-specific specialist models achieve high accuracy on singular objectives, such as property prediction or molecular generation, but require strict formatting and lack conversational flexibility. Conversely, early chemical large language models provide natural language interaction but are restricted to text and SMILES strings. ChemDFM-X addresses this gap by enabling large multimodal models to process the experimental characterization data (<a href="https://en.wikipedia.org/wiki/Tandem_mass_spectrometry">MS2 spectra</a> and <a href="https://en.wikipedia.org/wiki/Infrared_spectroscopy">IR spectra</a>) and visual data routinely used in practical chemistry workflows.</p>
<h2 id="synthetic-data-scaling-for-modality-alignment">Synthetic Data Scaling for Modality Alignment</h2>
<p>The core novelty lies in the <strong>&ldquo;Any-to-Text&rdquo; alignment strategy via synthetic data scaling</strong>:</p>
<ol>
<li>
<p><strong>Comprehensive Modality Support</strong>: ChemDFM-X incorporates experimental characterization data (MS2 and IR spectra) alongside 2D graphs, 3D conformations, and images. The data representations are formally defined mathematically rather than as raw pixels:</p>
<ul>
<li><strong>Molecular Graph</strong>: An undirected graph $G = (\textbf{V}, \textbf{E})$ with atom set $\textbf{V}$ and bond set $\textbf{E}$.</li>
<li><strong>Molecular Conformation</strong>: An undirected graph $G = (\textbf{V}&rsquo;, \textbf{E})$ storing spatial coordinates: $\textbf{v}_i = (x_i, y_i, z_i, a_i)$.</li>
<li><strong>MS2 Spectrum</strong>: Treated as a point sequence of discrete mass-to-charge ratios and intensities, tokenized via a discrete codebook: $\textbf{M} = ((r_1, I_1), (r_2, I_2), \dots, (r_n, I_n))$.</li>
<li><strong>IR Spectrum</strong>: Treated as a dense sequence of continuous wave lengths and absorption intensities, directly reshaped for feature extraction: $\textbf{R} = ((w_1, t_1), (w_2, t_2), \dots, (w_l, t_l))$.</li>
</ul>
<p>The authors trained new Sequence Transformer encoders from scratch for the MS2 and IR modalities since suitable pre-trained models did not exist.</p>
</li>
<li>
<p><strong>Synthetic Data Generation Pipeline</strong>: The authors generated a 7.6M sample dataset by starting with 1.3M seed SMILES and using &ldquo;approximate calculations&rdquo; to generate missing modalities:</p>
<ul>
<li>3D conformations via <a href="https://en.wikipedia.org/wiki/Merck_molecular_force_field">MMFF94</a> force field optimization</li>
<li>MS2 spectra via CFM-ID 4.0 (Competitive Fragmentation Modeling)</li>
<li>IR spectra via Chemprop-IR (Message Passing Neural Network)</li>
</ul>
</li>
<li>
<p><strong>Cross-Modal Synergy</strong>: The model demonstrates that training on reaction images improves recognition performance by leveraging semantic chemical knowledge (reaction rules) to correct visual recognition errors, an emergent capability from multimodal training.</p>
</li>
</ol>
<h2 id="multimodal-benchmarking-with-chemllmbench">Multimodal Benchmarking with ChemLLMBench</h2>
<p>The model was evaluated using a customized version of <strong><a href="/notes/chemistry/llm-applications/chemllmbench-eight-chemistry-tasks/">ChemLLMBench</a></strong> and <strong><a href="/notes/chemistry/molecular-design/property-prediction/moleculenet-benchmark-molecular-ml/">MoleculeNet</a></strong> across three modality categories:</p>
<ol>
<li>
<p><strong>Structural Modalities</strong> (2D Graphs &amp; 3D Conformations):</p>
<ul>
<li>Molecule recognition and captioning</li>
<li>Property prediction (MoleculeNet: BACE, BBBP, ClinTox, HIV, Tox21)</li>
<li>Compared against specialist models (Mole-BERT, Uni-Mol, MolXPT, MolCA) and generalist models (3D-MoLM, ChemDFM, <a href="/notes/chemistry/llm-applications/chemllm-chemical-large-language-model/">ChemLLM</a>)</li>
</ul>
</li>
<li>
<p><strong>Visual Modalities</strong> (Images):</p>
<ul>
<li>Single molecule image recognition</li>
<li>Reaction image recognition</li>
<li>Compared against GPT-4O, Gemini 1.5 Pro, Qwen-VL, LLaVA, and specialist models <a href="/notes/chemistry/optical-structure-recognition/vision-language/molnextr/">MolNextr</a> and <a href="/notes/chemistry/optical-structure-recognition/image-to-graph/molscribe/">MolScribe</a></li>
</ul>
</li>
<li>
<p><strong>Characterization Modalities</strong> (MS2 &amp; IR Spectra):</p>
<ul>
<li>Spectral analysis tasks (identifying molecules from spectra)</li>
<li>Contextualized spectral interpretation (combining spectra with reaction context)</li>
<li>Novel evaluation requiring integration of spectroscopic data with reaction knowledge</li>
</ul>
</li>
</ol>
<h2 id="cross-modal-synergy-and-generalist-performance">Cross-Modal Synergy and Generalist Performance</h2>
<p><strong>Key Findings</strong>:</p>
<ol>
<li>
<p><strong>Leading Generalist Performance</strong>: ChemDFM-X establishes a new benchmark among existing generalist models (such as 3D-MOLM and ChemLLM), achieving performance metrics that match dedicated specialist models across several multimodal tasks.</p>
</li>
<li>
<p><strong>Failure of General LMMs</strong>: General vision models (GPT-4O, Gemini 1.5 Pro, Qwen-VL, LLaVA, InternLM-XComposer2, DocOwl) failed significantly on chemical image recognition tasks (0% accuracy for most models on molecule and reaction recognition, Table 9), demonstrating that chemical domain knowledge cannot be assumed from general pre-training.</p>
</li>
<li>
<p><strong>Cross-Modal Error Correction</strong>: In reaction image recognition, ChemDFM-X achieved higher accuracy (53.0%) than on single molecules (46.0%) (Table 9). The authors conclude the model uses its internal knowledge of chemical reaction rules to correct recognition errors in the visual modality, an emergent capability from multimodal training.</p>
</li>
<li>
<p><strong>Reliance on Reaction Context for Spectra</strong>: In zero-shot scenarios, ChemDFM-X essentially fails at pure spectral recognition (achieving 0% and 1% top-1 accuracy on MS2 and IR spectra alone, Table 11). However, when SMILES-based reaction context is included, performance rises to 45% (MS2) and 64% (IR) on the reaction prediction task, and 29% (MS2) and 60% (IR) on <a href="https://en.wikipedia.org/wiki/Retrosynthetic_analysis">retrosynthesis</a> (Table 11). This indicates the model uses spectral data as a soft prior to constrain textual deductions. Furthermore, the paper compares ChemDFM-X’s spectral identification performance exclusively against text-only LLMs that cannot process spectra, omitting comparisons against established specialist tools.</p>
</li>
<li>
<p><strong>Surrogate Distillation Trade-offs</strong>: Because the spectral training data relies entirely on outputs from CFM-ID 4.0 and Chemprop-IR, ChemDFM-X effectively distills these surrogate models. Any inherent predictive biases or inaccuracies from these underlying tools are permanently embedded in the new ChemDFM-X encoders.</p>
</li>
</ol>
<p><strong>Main Conclusion</strong>: The &ldquo;separate encoders + unified decoder&rdquo; architecture with synthetic data generation enables effective multimodal chemical understanding, bridging the gap between specialist and generalist AI systems for chemistry.</p>
<hr>
<h2 id="reproducibility-details">Reproducibility Details</h2>
<h3 id="data">Data</h3>
<p>The authors constructed a <strong>7.6M sample instruction-tuning dataset</strong> derived from <strong>1.3M seed SMILES</strong> (sourced from <a href="https://en.wikipedia.org/wiki/PubChem">PubChem</a> and USPTO). <strong>Note</strong>: The final 7.6M multimodal tuning dataset itself isn&rsquo;t publicly available.</p>
<p><strong>Generation Pipeline</strong>:</p>
<table>
	<thead>
			<tr>
					<th>Modality</th>
					<th>Generation Method</th>
					<th>Tool/Model</th>
					<th>Sample Count</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><strong>2D Graphs</strong></td>
					<td>Direct extraction from SMILES</td>
					<td>RDKit</td>
					<td>1.1M</td>
			</tr>
			<tr>
					<td><strong>3D Conformations</strong></td>
					<td>Force field optimization</td>
					<td>RDKit + MMFF94</td>
					<td>1.3M (pseudo-optimal)</td>
			</tr>
			<tr>
					<td><strong>Molecule Images</strong></td>
					<td>Rendering with augmentation</td>
					<td>RDKit, Indigo, <a href="/notes/chemistry/optical-structure-recognition/hand-drawn/chempix/">ChemPix</a></td>
					<td>~1M (including handwritten style)</td>
			</tr>
			<tr>
					<td><strong>Reaction Images</strong></td>
					<td>Rendering from reaction SMILES</td>
					<td>RDKit</td>
					<td>300K</td>
			</tr>
			<tr>
					<td><strong>MS2 Spectra</strong></td>
					<td>Computational prediction</td>
					<td>CFM-ID 4.0</td>
					<td>~700K</td>
			</tr>
			<tr>
					<td><strong>IR Spectra</strong></td>
					<td>Computational prediction</td>
					<td>Chemprop-IR</td>
					<td>~1M</td>
			</tr>
	</tbody>
</table>
<p><strong>Data Augmentation</strong>:</p>
<ul>
<li>Molecule images augmented with &ldquo;handwritten&rdquo; style using the ChemPix pipeline</li>
<li>Multiple rendering styles (RDKit default, Indigo clean)</li>
<li>Spectra generated at multiple energy levels (10eV, 20eV, 40eV for MS2)</li>
</ul>
<h3 id="algorithms">Algorithms</h3>
<p><strong>Architecture</strong>: &ldquo;Separate Encoders + Unified Decoder&rdquo;</p>
<p><strong>Code Availability</strong>: The authors have only released inference logic. The cross-modal projection training and synthetic data-generation scripts are closed.</p>
<p><strong>Modality Alignment</strong>:</p>
<ul>
<li>Each modality has a dedicated encoder (frozen pre-trained models where available)</li>
<li>For graph, conformation, MS2, and IR modalities: <strong>2-layer MLP projector</strong> (Linear, GELU, Linear) maps encoder features to LLM input space</li>
<li>For images: <strong>H-Reducer</strong> module compresses image tokens by factor of $n=8$ to handle high-resolution chemical images, then projects to LLM input space</li>
<li>All projected features are concatenated and fed to the unified LLM decoder</li>
</ul>
<h3 id="models">Models</h3>
<p><strong>Base LLM</strong>:</p>
<ul>
<li><strong>ChemDFM (13B)</strong>: LLaMA-based model pre-trained on chemical text and SMILES</li>
</ul>
<p><strong>Modality Encoders</strong>:</p>
<table>
	<thead>
			<tr>
					<th>Modality</th>
					<th>Encoder</th>
					<th>Pre-training Data</th>
					<th>Parameter Count</th>
					<th>Status</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><strong>2D Graph</strong></td>
					<td>Mole-BERT</td>
					<td>2M molecules</td>
					<td>-</td>
					<td>Frozen</td>
			</tr>
			<tr>
					<td><strong>3D Conformation</strong></td>
					<td>Uni-Mol</td>
					<td>209M conformations</td>
					<td>-</td>
					<td>Frozen</td>
			</tr>
			<tr>
					<td><strong>Image</strong></td>
					<td>CLIP (ViT)</td>
					<td>General domain</td>
					<td>-</td>
					<td>Frozen</td>
			</tr>
			<tr>
					<td><strong>MS2 Spectrum</strong></td>
					<td>Transformer (SeqT)</td>
					<td>Trained from scratch</td>
					<td>-</td>
					<td><strong>Trainable</strong></td>
			</tr>
			<tr>
					<td><strong>IR Spectrum</strong></td>
					<td>Transformer (SeqT)</td>
					<td>Trained from scratch</td>
					<td>-</td>
					<td><strong>Trainable</strong></td>
			</tr>
	</tbody>
</table>
<p><strong>Design Rationale</strong>: MS2 and IR encoders trained from scratch as Sequence Transformers treating spectral peaks as token sequences, since no suitable pre-trained models exist for chemical spectra.</p>
<h3 id="evaluation">Evaluation</h3>
<p><strong>Metrics</strong>:</p>
<ul>
<li><strong>Accuracy (Acc)</strong> for recognition tasks</li>
<li><strong>BLEU-2/4</strong> and <strong>METEOR</strong> for captioning tasks</li>
<li><strong>AUC-ROC</strong> for property prediction (classification)</li>
</ul>
<p><strong>Code Availability</strong>: The adapted code for evaluating on ChemLLMBench and their custom spectral recognition tasks is closed-source.</p>
<p><strong>Benchmarks</strong>:</p>
<ul>
<li><strong>ChemLLMBench</strong>: Adapted for multimodal inputs across molecule captioning, property prediction, and reaction understanding</li>
<li><strong>MoleculeNet</strong>: Standard molecular property prediction tasks (BACE, BBBP, ClinTox, HIV, Tox21)</li>
<li><strong>USPTO</strong>: Reaction prediction and retrosynthesis tasks</li>
<li><strong>Custom Spectral Tasks</strong>: Novel evaluations requiring spectral interpretation</li>
</ul>
<h3 id="hardware">Hardware</h3>
<p><strong>Note</strong>: The type and quantity of GPUs used, along with the total training wall-time, were not published.</p>
<p><strong>Training Configuration</strong>:</p>
<ul>
<li><strong>Total Batch Size</strong>: 256</li>
<li><strong>Epochs</strong>: 3</li>
<li><strong>Optimizer</strong>: AdamW</li>
</ul>
<p><strong>Modality-Specific Learning Rates (Peak)</strong>:</p>
<table>
	<thead>
			<tr>
					<th>Modality</th>
					<th>Learning Rate</th>
					<th>Feature Dimension</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td>Graph</td>
					<td>1e-5</td>
					<td>300</td>
			</tr>
			<tr>
					<td>Conformation</td>
					<td>2e-4</td>
					<td>512</td>
			</tr>
			<tr>
					<td>Image</td>
					<td>2e-3</td>
					<td>1024</td>
			</tr>
			<tr>
					<td>MS2 / IR</td>
					<td>2e-4</td>
					<td>768</td>
			</tr>
	</tbody>
</table>
<p><strong>Note</strong>: Different learning rates reflect the varying degrees of domain adaptation required. Images (general CLIP) need more adaptation than graphs (chemical Mole-BERT).</p>
<h3 id="artifacts">Artifacts</h3>
<table>
	<thead>
			<tr>
					<th>Artifact</th>
					<th>Type</th>
					<th>License</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><a href="https://github.com/OpenDFM/ChemDFM-X">ChemDFM-X (GitHub)</a></td>
					<td>Code</td>
					<td>Apache-2.0</td>
					<td>Inference code only; training and data generation scripts are closed</td>
			</tr>
			<tr>
					<td><a href="https://huggingface.co/OpenDFM/ChemDFM-X-v1.0-13B">ChemDFM-X-v1.0-13B (HuggingFace)</a></td>
					<td>Model</td>
					<td>AGPL-3.0</td>
					<td>13B parameter multimodal model weights</td>
			</tr>
	</tbody>
</table>
<hr>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation</strong>: Zhao, Z., Chen, B., Li, J., Chen, L., Wen, L., Wang, P., Zhu, Z., Zhang, D., Wan, Z., Li, Y., Dai, Z., Chen, X., &amp; Yu, K. (2024). ChemDFM-X: Towards Large Multimodal Model for Chemistry. <em>Science China Information Sciences</em>, 67(12), 220109. <a href="https://doi.org/10.1007/s11432-024-4243-0">https://doi.org/10.1007/s11432-024-4243-0</a></p>
<p><strong>Publication</strong>: Science China Information Sciences, December 2024</p>
<p><strong>Additional Resources</strong>:</p>
<ul>
<li><a href="https://arxiv.org/abs/2409.13194">arXiv Version</a></li>
<li><a href="https://github.com/OpenDFM/ChemDFM-X">Code Repository</a></li>
</ul>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@article</span>{zhaoChemDFMXLargeMultimodal2024,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span> = <span style="color:#e6db74">{{{ChemDFM-X}}: {{Towards Large Multimodal Model}} for {{Chemistry}}}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span> = <span style="color:#e6db74">{Zhao, Zihan and Chen, Bo and Li, Jingpiao and Chen, Lu and Wen, Liyang and Wang, Pengyu and Zhu, Zichen and Zhang, Danyang and Wan, Ziping and Li, Yansi and Dai, Zhongyang and Chen, Xin and Yu, Kai}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span> = <span style="color:#e6db74">{2024}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">month</span> = dec,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">journal</span> = <span style="color:#e6db74">{Science China Information Sciences}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">volume</span> = <span style="color:#e6db74">{67}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">number</span> = <span style="color:#e6db74">{12}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">pages</span> = <span style="color:#e6db74">{220109}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">doi</span> = <span style="color:#e6db74">{10.1007/s11432-024-4243-0}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">archiveprefix</span> = <span style="color:#e6db74">{arXiv}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">eprint</span> = <span style="color:#e6db74">{2409.13194}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">primaryclass</span> = <span style="color:#e6db74">{cs.LG}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div>]]></content:encoded></item><item><title>ChemVLM: A Multimodal Large Language Model for Chemistry</title><link>https://hunterheidenreich.com/notes/chemistry/llm-applications/chemvlm/</link><pubDate>Fri, 19 Dec 2025 00:00:00 +0000</pubDate><guid>https://hunterheidenreich.com/notes/chemistry/llm-applications/chemvlm/</guid><description>A 26B parameter multimodal LLM for chemistry, combining InternViT-6B and ChemLLM-20B for molecular structure recognition, property prediction, and reasoning.</description><content:encoded><![CDATA[<h2 id="paper-classification-method-and-resource">Paper Classification: Method and Resource</h2>
<p>This paper is a combination of <strong>Method</strong> (primary) and <strong>Resource</strong> (secondary).</p>
<p>It is primarily a <strong>Method</strong> paper because it proposes <strong>ChemVLM</strong>, a novel multimodal architecture specifically tailored for the chemical domain, utilizing a &ldquo;ViT-MLP-LLM&rdquo; framework. The authors introduce a specific two-stage training strategy to align visual features with chemical text representations.</p>
<p>Secondarily, it is a <strong>Resource</strong> paper as it introduces a comprehensive suite of three new datasets: <strong>ChemOCR</strong>, <strong>MMCR-Bench</strong>, and <strong>MMChemBench</strong>, developed to rigorously evaluate multimodal capabilities in chemistry, covering OCR, reasoning, and property prediction.</p>
<h2 id="bridging-the-visual-gap-in-chemical-llms">Bridging the Visual Gap in Chemical LLMs</h2>
<p>The primary motivation is the limitation of existing models in handling the multimodal nature of chemistry.</p>
<ul>
<li><strong>Visual Data Gap</strong>: Chemical tasks heavily rely on visual information (molecular structures, reactions) which purely text-based chemical LLMs cannot process.</li>
<li><strong>Limitations of Generalist Models</strong>: General multimodal models (like GPT-4V or LLaVA) lack specialized chemical domain knowledge, leading to hallucinations or misinterpretations.</li>
<li><strong>Inadequacy of OCR Tools</strong>: Traditional <a href="/notes/chemistry/optical-structure-recognition/">chemical OCR</a> tools (like <a href="/notes/chemistry/optical-structure-recognition/image-to-graph/molscribe/">MolScribe</a>) excel at modality conversion (Image-to-<a href="/notes/chemistry/molecular-representations/notations/smiles/">SMILES</a>) but fail at complex reasoning tasks.</li>
</ul>
<h2 id="domain-specific-data-curation-and-benchmarking">Domain-Specific Data Curation and Benchmarking</h2>
<ul>
<li><strong>Data-Driven Alignment</strong>: The underlying &ldquo;ViT-MLP-LLM&rdquo; framework is standard in multimodal modeling, paralleling architectures like LLaVA. The core innovation here is the rigorous creation of a bilingual multimodal dataset spanning hand-drawn molecules, reactions, and exam questions augmented with style transfers. The training data pipeline heavily relies on generating synthetic variance using tools like RanDepict and <a href="https://en.wikipedia.org/wiki/RDKit">RDKit</a> to introduce distortions, rotations, and handwritten styles, alongside GPT-4 generated prompts to ensure linguistic diversity.</li>
<li><strong>Model Integration</strong>: ChemVLM merges <strong>InternViT-6B</strong> (a large-scale vision transformer) with <strong><a href="/notes/chemistry/llm-applications/chemllm-chemical-large-language-model/">ChemLLM-20B</a></strong> (a chemical language model). Visual features $X_v$ are mapped into the linguistic embedding space via an MLP projector, producing aligned token sequences alongside text instructions $X_q$. The joint multimodal sequence is trained using standard autoregressive next-token prediction:
$$ \mathcal{L} = -\sum_{i} \log P(y_i \mid X_v, X_q, y_{&lt;i}) $$</li>
<li><strong>Three Custom Benchmarks</strong>: The authors introduce tailored benchmarks to assess distinct competencies:
<ul>
<li><strong>ChemOCR</strong>: For image-to-SMILES conversion.</li>
<li><strong>MMCR-Bench</strong>: College entrance exam questions testing complex logical reasoning.</li>
<li><strong>MMChemBench</strong>: For molecule captioning and zero-shot property prediction.</li>
</ul>
</li>
</ul>
<h2 id="evaluating-chemical-ocr-and-reasoning">Evaluating Chemical OCR and Reasoning</h2>
<p>The authors benchmarked ChemVLM against both open-source (LLaVA, Qwen-VL, InternVL) and proprietary (GPT-4V) models across three primary domains:</p>
<ol>
<li><strong>Chemical OCR</strong>: Evaluated on 1,000 image-text pairs from ChemOCR. The primary metric is the <a href="https://en.wikipedia.org/wiki/Jaccard_index">Tanimoto similarity</a> between the Morgan fingerprints of the generated structure ($A$) and the ground-truth SMILES ($B$):
$$ T(A, B) = \frac{|A \cap B|}{|A| + |B| - |A \cap B|} $$
They report both the average Tanimoto similarity and the strict exact-match rate (<code>Tanimoto@1.0</code>).</li>
<li><strong>Multimodal Chemical Reasoning (MMCR)</strong>: Tested on MMCR-Bench (1,000 exam questions), ScienceQA, and CMMU. Performance was scored based on accuracy for multiple-choice and fill-in-the-blank questions.</li>
<li><strong>Multimodal Molecule Understanding</strong>: Evaluated on MMChemBench for molecule captioning and property prediction.</li>
<li><strong>Text-Only Reasoning</strong>: Tested on SciBench, a text-only benchmark for university-level science, to ensure the model retains fundamental linguistic reasoning.</li>
<li><strong>Generalization</strong>: Tested on non-chemistry subjects within the CMMU framework (Biology, Physics, Math) to assess cross-domain competence.</li>
</ol>
<h2 id="performance-gains-and-existing-limitations">Performance Gains and Existing Limitations</h2>
<ul>
<li><strong>Multimodal Reasoning Leadership</strong>: ChemVLM achieved state-of-the-art results on MMCR-Bench (41.7%), surpassing generalist models like GPT-4V (40.1%). However, scoring for portions of these benchmarks relied heavily on an LLM-as-a-judge (the Qwen-max API), which can introduce bias as LLM evaluators often favor structural characteristics and verbosity produced by similar autoregressive models. Furthermore, the model was fine-tuned on 200,000 exam questions and tested on MMCR-Bench (also derived from Chinese college entrance exams). While the authors state the data was deduplicated, the potential for data leakage remains a significant unaddressed confounder.</li>
<li><strong>Superior Understanding</strong>: In molecule captioning and prediction, ChemVLM showed significant improvements over general baseline models, scoring 80.9% on prediction compared to GPT-4V&rsquo;s 38.6%. This is a natural consequence of testing a custom-trained model on domain-specific benchmarks.</li>
<li><strong>OCR Capabilities vs. Dedicated Tools</strong>: ChemVLM outperformed generalist MLLMs in chemical structure recognition, achieving an average Tanimoto similarity of 71.0% (vs. GPT-4V&rsquo;s 15.0%). However, it remains significantly inferior to pure structural OCR tools like MolScribe in strict modality conversion tasks, only achieving an exact structural match (<code>Tanimoto@1.0</code>) of 42.9% compared to MolScribe&rsquo;s 89.1%.</li>
<li><strong>Textual Retention and Generalization Claims</strong>: The authors claim the diverse training strategy imparts broad scientific reasoning, pointing to performance retention on non-chemistry subjects (Biology, Physics, Math) and strong results on the purely textual SciBench benchmark. However, this cross-domain generalization highly likely stems from the underlying base model (ChemLLM-20B/InternLM2) or the inclusion of 1.3 million &ldquo;General&rdquo; visual QA pairs in their training blend, rather than emergent general scientific skills originating purely from learning chemistry representations.</li>
</ul>
<hr>
<h2 id="reproducibility-details">Reproducibility Details</h2>
<h3 id="data">Data</h3>
<p>The training and evaluation data relied on a mix of open-source repositories and custom curation. Many of the curated datasets have been formally released by the authors on Hugging Face (<a href="https://huggingface.co/datasets/di-zhang-fdu/chemvlm-sft-datasets"><code>di-zhang-fdu/chemvlm-sft-datasets</code></a>).</p>
<table>
	<thead>
			<tr>
					<th>Purpose</th>
					<th>Dataset</th>
					<th>Source/Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><strong>Training (Molecule)</strong></td>
					<td><strong><a href="/notes/chemistry/optical-structure-recognition/hand-drawn/decimer-hand-drawn/">DECIMER HDM</a></strong></td>
					<td>7,000+ hand-drawn molecular images.</td>
			</tr>
			<tr>
					<td><strong>Training (Molecule)</strong></td>
					<td><strong>MolScribe Data</strong></td>
					<td>Scanned/photographed images from literature.</td>
			</tr>
			<tr>
					<td><strong>Training (Molecule)</strong></td>
					<td><strong>Synthetic</strong></td>
					<td>Generated via ChemDraw, RDKit, and Indigo with style transfer (blurring, rotation, handwritten styles).</td>
			</tr>
			<tr>
					<td><strong>Training (Reaction)</strong></td>
					<td><strong>PEACE &amp; USPTO-50K</strong></td>
					<td>Inorganic and organic reaction schemes.</td>
			</tr>
			<tr>
					<td><strong>Training (Reasoning)</strong></td>
					<td><strong>Exam Questions</strong></td>
					<td>200,000 questions from OpenDataLab (Chinese education level). <a href="https://huggingface.co/collections/di-zhang-fdu/multi-corpus-datasets-for-chemllm">Available on Hugging Face</a>.</td>
			</tr>
			<tr>
					<td><strong>Evaluation</strong></td>
					<td><strong>ChemOCR</strong></td>
					<td>1,000 bilingual image-text pairs for SMILES recognition. Released via Google Drive link in repo.</td>
			</tr>
			<tr>
					<td><strong>Evaluation</strong></td>
					<td><strong>MMCR-Bench</strong></td>
					<td>1,000 multimodal chemistry exam questions. <strong>Requires emailing authors directly for access.</strong></td>
			</tr>
			<tr>
					<td><strong>Evaluation</strong></td>
					<td><strong>MMChemBench</strong></td>
					<td>Extension of <a href="/notes/chemistry/llm-applications/chembench-llm-chemistry-evaluation/">ChemBench</a> for captioning and property prediction. Released via Google Drive link in repo.</td>
			</tr>
	</tbody>
</table>
<p><strong>Preprocessing</strong>: Images were augmented using <strong>RanDepict</strong> for style variation. Text data (SMILES) was validated and cleaned. Prompts were diversified using GPT-4 to generate different linguistic styles.</p>
<h3 id="algorithms">Algorithms</h3>
<ul>
<li><strong>Architecture</strong>: &ldquo;ViT-MLP-LLM&rdquo; structure.
<ul>
<li><strong>Vision Encoder</strong>: InternViT-6B, processing images at $448 \times 448$ resolution. Images are segmented into tiles (max 12).</li>
<li><strong>Projector</strong>: Multi-Layer Perceptron (MLP) initialized randomly to map visual features to text embedding space.</li>
<li><strong>LLM</strong>: ChemLLM-20B, a domain-specific model.</li>
</ul>
</li>
<li><strong>Training Strategy</strong>: Two-stage supervised fine-tuning.
<ol>
<li><strong>Modal Alignment</strong>: Freeze LLM and base Vision Encoder weights. Train only the randomly initialized MLP projector and LoRA layers (rank 32) of the Vision Encoder. Uses diverse multimodal data.</li>
<li><strong>Supervised Fine-Tuning (SFT)</strong>: Keep LLM and Vision Encoder base weights frozen, but add LoRA (rank 16) to the LLM and retain LoRA (rank 32) on the Vision Encoder. The MLP projector is fully trained. Data includes specialized chemistry and general corpora.</li>
</ol>
</li>
<li><strong>Optimization</strong>:
<ul>
<li>Optimizer: AdamW</li>
<li>Context Length: 2048 tokens</li>
<li>Chat Template: InternLM2 dialogue schema</li>
</ul>
</li>
</ul>
<h3 id="models">Models</h3>
<ul>
<li><strong>ChemVLM-26B</strong>: The primary model released. It combines the 6B parameter vision encoder and the 20B parameter language model. Weights are fully available at <a href="https://huggingface.co/AI4Chem/ChemVLM-26B-1-2"><code>AI4Chem/ChemVLM-26B-1-2</code></a>. An 8B version is also available.</li>
<li><strong>Baselines</strong>: Comparisons were made against <strong>GPT-4V</strong>, <strong>Qwen-VL-Chat</strong>, <strong>LLaVA-v1.5-13B</strong>, <strong>InternVL-v1.5</strong>, and <strong>Yi-VL-Plus</strong>.</li>
</ul>
<h3 id="evaluation">Evaluation</h3>
<p>Performance was measured across three distinct task types. Exact <a href="https://github.com/lijunxian111/ChemVlm/tree/master/evaluation">evaluation scripts</a> have been released in the official repository.</p>
<table>
	<thead>
			<tr>
					<th>Metric</th>
					<th>Task</th>
					<th>Method</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><strong>Tanimoto Similarity</strong></td>
					<td>ChemOCR</td>
					<td>Comparison of generated SMILES vs. ground truth using RDKit. Reports Average Similarity and <code>Tanimoto@1.0</code> (exact match).</td>
			</tr>
			<tr>
					<td><strong>Accuracy</strong></td>
					<td>MMCR (Reasoning)</td>
					<td>+1 point for correct multiple-choice/fill-in-the-blank; 0 otherwise. Scored via Qwen-max API prompting.</td>
			</tr>
			<tr>
					<td><strong>Prediction Score</strong></td>
					<td>Property Prediction</td>
					<td>Evaluated on MMChemBench subsets.</td>
			</tr>
	</tbody>
</table>
<h3 id="hardware">Hardware</h3>
<ul>
<li><strong>Training Compute</strong>: Training utilized <strong>16 NVIDIA A100 (80GB)</strong> GPUs.</li>
<li><strong>Configuration</strong>:
<ul>
<li>Batch size: 4 (per GPU, resulting in an effective global batch size of 256)</li>
<li>Gradient Accumulation: 4 iterations</li>
<li>Precision: <strong><a href="https://en.wikipedia.org/wiki/DeepSpeed">Deepspeed</a> bfloat16 (bf16)</strong> with <strong>ZeRO-3</strong> offloading strategy</li>
<li>Framework: Training runs on the InternVL-v1.5 codebase rather than standalone scripts.</li>
</ul>
</li>
<li><strong>Inference Compute</strong>: Evaluating the 26B model requires at least one 80GB A100 GPU (with Flash Attention + bfloat16). The 8B variant requires a GPU with at least 48GB of VRAM.</li>
</ul>
<h3 id="artifacts">Artifacts</h3>
<table>
	<thead>
			<tr>
					<th>Artifact</th>
					<th>Type</th>
					<th>License</th>
					<th>Notes</th>
			</tr>
	</thead>
	<tbody>
			<tr>
					<td><a href="https://huggingface.co/AI4Chem/ChemVLM-26B">ChemVLM-26B</a></td>
					<td>Model</td>
					<td>MIT</td>
					<td>Original 26B model weights</td>
			</tr>
			<tr>
					<td><a href="https://huggingface.co/AI4Chem/ChemVLM-26B-1-2">ChemVLM-26B-1-2</a></td>
					<td>Model</td>
					<td>Apache-2.0</td>
					<td>Updated 26B model weights</td>
			</tr>
			<tr>
					<td><a href="https://huggingface.co/datasets/di-zhang-fdu/chemvlm-sft-datasets">chemvlm-sft-datasets</a></td>
					<td>Dataset</td>
					<td>Unknown</td>
					<td>SFT training data (~51.7k rows)</td>
			</tr>
			<tr>
					<td><a href="https://github.com/lijunxian111/ChemVlm">ChemVlm (GitHub)</a></td>
					<td>Code</td>
					<td>Unknown</td>
					<td>Training, evaluation, and inference code</td>
			</tr>
	</tbody>
</table>
<h2 id="paper-information">Paper Information</h2>
<p><strong>Citation</strong>: Li, J., et al. (2025). ChemVLM: Exploring the Power of Multimodal Large Language Models in Chemistry Area. <em>Proceedings of the AAAI Conference on Artificial Intelligence</em>, 39(1), 415-423. <a href="https://doi.org/10.1609/aaai.v39i1.32020">https://doi.org/10.1609/aaai.v39i1.32020</a></p>
<p><strong>Publication</strong>: AAAI 2025</p>
<div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;-webkit-text-size-adjust:none;"><code class="language-bibtex" data-lang="bibtex"><span style="display:flex;"><span><span style="color:#a6e22e">@inproceedings</span>{li2025chemvlm,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">title</span>=<span style="color:#e6db74">{ChemVLM: Exploring the Power of Multimodal Large Language Models in Chemistry Area}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">author</span>=<span style="color:#e6db74">{Li, Junxian and Zhang, Di and Wang, Xunzhi and Hao, Zeying and Lei, Jingdi and Tan, Qian and Zhou, Cai and Liu, Wei and Yang, Yaotian and Xiong, Xinrui and Wang, Weiyun and Chen, Zhe and Wang, Wenhai and Li, Wei and Su, Mao and Zhang, Shufei and Ouyang, Wanli and Li, Yuqiang and Zhou, Dongzhan}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">booktitle</span>=<span style="color:#e6db74">{Proceedings of the AAAI Conference on Artificial Intelligence}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">volume</span>=<span style="color:#e6db74">{39}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">number</span>=<span style="color:#e6db74">{1}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">pages</span>=<span style="color:#e6db74">{415--423}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">year</span>=<span style="color:#e6db74">{2025}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">url</span>=<span style="color:#e6db74">{https://doi.org/10.1609/aaai.v39i1.32020}</span>,
</span></span><span style="display:flex;"><span>  <span style="color:#a6e22e">doi</span>=<span style="color:#e6db74">{10.1609/aaai.v39i1.32020}</span>
</span></span><span style="display:flex;"><span>}
</span></span></code></pre></div><p><strong>Additional Resources</strong>:</p>
<ul>
<li><a href="https://github.com/lijunxian111/ChemVlm">Official Repository</a></li>
</ul>
]]></content:encoded></item></channel></rss>