S3091 Multimodal Transformer Model Predicts Post-Treatment Lesion Recurrence in Gastrointestinal Patients

Introduction: Accurate prediction of gastrointestinal lesion recurrence after treatment remains a critical unmet need. Integrating visual features from endoscopic imaging with patient clinical data may enhance risk stratification. We developed and evaluated a multimodal deep‐learning pipeline combining Vision Transformer embeddings of capsule endoscopy images with structured metadata to predict lesion recurrence. Methods: We retrospectively assembled a cohort of 1,200 adult patients from the KYUCapsule dataset who underwent capsule endoscopy and follow-up for at least 12 months. We annotated 3,600 high-resolution lesion frames, categorizing each by lesion type (inflammatory, vascular, or neoplastic) and extracting corresponding metadata: age, sex, treatment regimen, lesion classification, history of prior recurrence, and follow-up duration. Image frames passed through a pre trained Vision Transformer (ViT), with final layers fine-tuned on our dataset to produce 512-dimension feature vectors capturing morphology and texture. Simultaneously, we encoded structured metadata via a fully connected neural network (128 and 64 neurons, ReLU activation). We concatenated image and metadata embeddings and trained a multilayer perceptron classifier with 2 hidden layers (256 and 128 neurons) and dropout (rate = 0.3). We split data into training (70%), validation (15%), and test (15%) sets, ensuring equal representation of recurrence vs non-recurrence cases through oversampling. Hyperparameters (learning rate, batch size, L2 regularization λ = 0.001) were optimized via grid search on the validation set. We used cross-entropy loss, Adam optimization, and early stopping. Results: On the test set (n = 180), the multimodal model achieved accuracy = 89.2%, precision = 88.7%, recall = 88.5%, and F1-score = 88.9%. The area under the ROC curve reached 0.902. Subgroup analysis showed consistent accuracy across lesion types: inflammatory = 90.5%, vascular = 88.0%, and neoplastic = 89.8%. Calibration was strong (Brier score = 0.087). SHAP analysis identified lesion type, prior recurrence history, and image-derived severity metrics as the top 3 predictors. Conclusion: Our multimodal transformer-based approach reliably integrates endoscopic imaging and clinical metadata to predict gastrointestinal lesion recurrence. Given its high discriminative performance and interpretability, this model could aid clinicians in tailoring surveillance and management strategies, potentially improving patient outcomes.

Publication Details

Journal
The American Journal of Gastroenterology
Published
2025-10-01
DOI
https://doi.org/10.14309/01.ajg.0001139824.61211.24
Primary Topic
Gastrointestinal Bleeding Diagnosis and Treatment
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

S3091 Multimodal Transformer Model Predicts Post-Treatment Lesion Recurrence in Gastrointestinal Patients

The American Journal of Gastroenterology
Gastrointestinal Bleeding Diagnosis and Treatment
article

S3091 Multimodal Transformer Model Predicts Post-Treatment Lesion Recurrence in Gastrointestinal Patients

article en

Abstract

Introduction: Accurate prediction of gastrointestinal lesion recurrence after treatment remains a critical unmet need. Integrating visual features from endoscopic imaging with patient clinical data may enhance risk stratification. We developed and evaluated a multimodal deep‐learning pipeline combining Vision Transformer embeddings of capsule endoscopy images with structured metadata to predict lesion recurrence. Methods: We retrospectively assembled a cohort of 1,200 adult patients from the KYUCapsule dataset who underwent capsule endoscopy and follow-up for at least 12 months. We annotated 3,600 high-resolution lesion frames, categorizing each by lesion type (inflammatory, vascular, or neoplastic) and extracting corresponding metadata: age, sex, treatment regimen, lesion classification, history of prior recurrence, and follow-up duration. Image frames passed through a pre trained Vision Transformer (ViT), with final layers fine-tuned on our dataset to produce 512-dimension feature vectors capturing morphology and texture. Simultaneously, we encoded structured metadata via a fully connected neural network (128 and 64 neurons, ReLU activation). We concatenated image and metadata embeddings and trained a multilayer perceptron classifier with 2 hidden layers (256 and 128 neurons) and dropout (rate = 0.3). We split data into training (70%), validation (15%), and test (15%) sets, ensuring equal representation of recurrence vs non-recurrence cases through oversampling. Hyperparameters (learning rate, batch size, L2 regularization λ = 0.001) were optimized via grid search on the validation set. We used cross-entropy loss, Adam optimization, and early stopping. Results: On the test set (n = 180), the multimodal model achieved accuracy = 89.2%, precision = 88.7%, recall = 88.5%, and F1-score = 88.9%. The area under the ROC curve reached 0.902. Subgroup analysis showed consistent accuracy across lesion types: inflammatory = 90.5%, vascular = 88.0%, and neoplastic = 89.8%. Calibration was strong (Brier score = 0.087). SHAP analysis identified lesion type, prior recurrence history, and image-derived severity metrics as the top 3 predictors. Conclusion: Our multimodal transformer-based approach reliably integrates endoscopic imaging and clinical metadata to predict gastrointestinal lesion recurrence. Given its high discriminative performance and interpretability, this model could aid clinicians in tailoring surveillance and management strategies, potentially improving patient outcomes.

The American Journal of GastroenterologyVol. 120(10S2)
Openalex Percentile: Top 29%
Gastrointestinal Bleeding Diagnosis and Treatment
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.