From Pixels to Pathology: Restoration Diffusion for Diagnostic-Consistent Virtual IHC

2508.02528v1 eess.IV, cs.CV 2025-08-09
Авторы:

Jingsong Liu, Xiaofeng Deng, Han Li, Azar Kazemi, Christian Grashei, Gesa Wilkens, Xin You, Tanja Groll, Nassir Navab, Carolin Mogler, Peter J. Schüffler

Резюме на русском

Оценка маркеров в клетках через иммуногистохимический анализ (IHC) важно для диагностики и лечения рака, но этот процесс затратен и требует много времени. Альтернативой является виртуальное статирование — трансформация типовых H&E-выкладок в IHC-выкладки с помощью искусственных нейронных сетей. Однако существуют две проблемы: необходимость сравнения синтетических изображений с неидеальными IHC-оригиналами и сохранение биологической разнообразности при преобразовании. Мы предлагаем Star-Diff, модель diffusion-based, которая решает эти задачи, сохраняя ткачественную структуру и моделируя видоизменения маркеров. Для оценки результатов мы предложили Semantic Fidelity Score (SFS), метрику, основанную на классификации маркеров, которая эффективна даже при недостаточном выравнивании изображений. Star-Diff показала лучшие результаты в SOTA на датасете BCI, демонстрируя промышленную применимость для виртуального IHC во время операции.

Abstract

Hematoxylin and eosin (H&E) staining is the clinical standard for assessing tissue morphology, but it lacks molecular-level diagnostic information. In contrast, immunohistochemistry (IHC) provides crucial insights into biomarker expression, such as HER2 status for breast cancer grading, but remains costly and time-consuming, limiting its use in time-sensitive clinical workflows. To address this gap, virtual staining from H&E to IHC has emerged as a promising alternative, yet faces two core challenges: (1) Lack of fair evaluation of synthetic images against misaligned IHC ground truths, and (2) preserving structural integrity and biological variability during translation. To this end, we present an end-to-end framework encompassing both generation and evaluation in this work. We introduce Star-Diff, a structure-aware staining restoration diffusion model that reformulates virtual staining as an image restoration task. By combining residual and noise-based generation pathways, Star-Diff maintains tissue structure while modeling realistic biomarker variability. To evaluate the diagnostic consistency of the generated IHC patches, we propose the Semantic Fidelity Score (SFS), a clinical-grading-task-driven metric that quantifies class-wise semantic degradation based on biomarker classification accuracy. Unlike pixel-level metrics such as SSIM and PSNR, SFS remains robust under spatial misalignment and classifier uncertainty. Experiments on the BCI dataset demonstrate that Star-Diff achieves state-of-the-art (SOTA) performance in both visual fidelity and diagnostic relevance. With rapid inference and strong clinical alignment,it presents a practical solution for applications such as intraoperative virtual IHC synthesis.

Ссылки и действия