RepViT-CXR: A Channel Replication Strategy for Vision Transformers in Chest X-ray Tuberculosis and Pneumonia Classification
2509.08234v1
cs.CV, cs.LG, F.2.2; I.2.7
2025-09-11
Авторы:
Faisal Ahmed
Резюме на русском
#### Контекст
Обработка зрительных снимков, таких как рентгеновские снимки (CXR), является ключевым инструментом в диагностике многих заболеваний, в том числе туберкулеза (TB) и нозокомиальной пневмонии. Несмотря на то, что традиционные методы машинного обучения, такие как действующие поля (CNN), достигли значительных успехов в этой области, последние развития в области трансформеров (ViT) показали свое потенциал для автоматизированного анализа медицинских изображений. Однако, существующие ViTs, предназначенные для трех-канальных изображений (RGB), не могут напрямую обрабатывать отдельные (grayscale) CXR снимки без дополнительных модификаций. Это ограничение может привести к неэффективному использованию моделей. Данное исследование направлено на решение этой проблемы, предлагая стратегию репликации каналов (RepViT-CXR), которая позволяет адаптировать CXR к требованиям ViTs без потери информации.
#### Метод
RepViT-CXR предлагает простую, но эффективную стратегию для адаптации CXR к ViT-моделям. Основная идея заключается в репликации сигнала канала (сглаживание значений пикселей) для создания имитированного трех-канального формата, который может быть просто вставлен в стандартную архитектуру ViT. Эта стратегия позволяет использовать существующие ViTs, не требуя изменений в их основной иерархии или предварительной обработке. Для оценки модели использованы данные из трех открытых баз данных: TB-CXR, Pediatric Pneumonia и Shenzhen TB. Методы оценки включали различные метрики, такие как точность, перенос, значимость и AUC.
#### Результаты
На TB-CXR датасете, RepViT-CXR достигла ошибки в диагностике 0.1% (accuracy 99.9%) и AUC 99.9%, превосходя лучшие результаты предыдущих работ (Topo-CXR, 99.3% accuracy, 99.8% AUC). На Pediatric Pneumonia данные показали accuracy 99.0%, recall 99.2%, precision 99.3% и AUC 99.0%, намного превосходя DCNN и VGG16. На Shenzhen TB датасете, RepViT-CXR показала accuracy 91.1% и AUC 91.2%, значительно улучшив результаты выдачи предыдущих CNN-based моделей. Эти результаты подтверждают высокую эффективность RepViT-CXR в классификации туберкулеза и пневмонии на основе CXR.
#### Значимость
RepViT-CXR может быть применена в области диагностики TB и пневмонии, существенно повысив точность и скорость таких процессов. Также, модель доказала свою эффективность на разных датасетах, что делает её универсальным инструментом для медицинского анализа. Помимо этого, RepViT-CXR может быть использована в разработке более широкого класса моделей для других медицинских изображений, требующих адаптации ViT к градациям с
Abstract
Chest X-ray (CXR) imaging remains one of the most widely used diagnostic
tools for detecting pulmonary diseases such as tuberculosis (TB) and pneumonia.
Recent advances in deep learning, particularly Vision Transformers (ViTs), have
shown strong potential for automated medical image analysis. However, most ViT
architectures are pretrained on natural images and require three-channel
inputs, while CXR scans are inherently grayscale. To address this gap, we
propose RepViT-CXR, a channel replication strategy that adapts single-channel
CXR images into a ViT-compatible format without introducing additional
information loss. We evaluate RepViT-CXR on three benchmark datasets. On the
TB-CXR dataset,our method achieved an accuracy of 99.9% and an AUC of 99.9%,
surpassing prior state-of-the-art methods such as Topo-CXR (99.3% accuracy,
99.8% AUC). For the Pediatric Pneumonia dataset, RepViT-CXR obtained 99.0%
accuracy, with 99.2% recall, 99.3% precision, and an AUC of 99.0%,
outperforming strong baselines including DCNN and VGG16. On the Shenzhen TB
dataset, our approach achieved 91.1% accuracy and an AUC of 91.2%, marking a
performance improvement over previously reported CNN-based methods. These
results demonstrate that a simple yet effective channel replication strategy
allows ViTs to fully leverage their representational power on grayscale medical
imaging tasks. RepViT-CXR establishes a new state of the art for TB and
pneumonia detection from chest X-rays, showing strong potential for deployment
in real-world clinical screening systems.