MGFormer: Super-Resolution Reconstruction of Retinal OCT Images Based on a Multi-Granularity Transformer
Optical coherence tomography (OCT) acquisitions often reduce lateral sampling density to shorten scan time and suppress motion artifacts, but this strategy degrades the signal-to-noise ratio and obscures fine retinal microstructures. To recover these details without hardware modifications, we propose MGFormer, a lightweight Transformer for OCT super-resolution (SR) that integrates a multi-granularity attention mechanism with tensor distillation. A feature-enhancing convolution first sharpens edges; stacked multi-granularity attention blocks then fuse coarse-to-fine context, while a row-wise top-k operator retains the most informative tokens and preserves their positional order. We trained and evaluated MGFormer on B-scans from the Duke SD-OCT dataset at 2×, 4×, and 8× scaling factors. Relative to seven recent CNN- and Transformer-based SR models, MGFormer achieves the highest quantitative fidelity; at 4× it reaches 34.39 dB PSNR and 0.8399 SSIM, surpassing SwinIR by +0.52 dB and +0.026 SSIM, and reduces LPIPS by 21.4%. Compared with the same backbone without tensor distillation, FLOPs drop from 289G to 233G (−19.4%), and per-B-scan latency at 4× falls from 166.43 ms to 98.17 ms (−41.01%); the model size remains compact (105.68 MB). A blinded reader study shows higher scores for boundary sharpness (4.2 ± 0.3), pathology discernibility (4.1 ± 0.3), and diagnostic confidence (4.3 ± 0.2), exceeding SwinIR by 0.3–0.5 points. These results suggest that MGFormer can provide fast, high-fidelity OCT SR suitable for routine clinical workflows.
Authors
- Yichao Si (ORCID: https://orcid.org/0009-0000-6608-3492)
- Jingmin Luan (ORCID: https://orcid.org/0000-0003-1476-0732)
- Jia Sun (ORCID: https://orcid.org/0000-0002-9188-7000)
- Zhenhe Ma (ORCID: https://orcid.org/0000-0003-4079-9550)
- Dongni Yang
- Yutian Li (ORCID: https://orcid.org/0000-0003-1810-3000)
- Yao Yu (ORCID: https://orcid.org/0000-0002-9754-5597)
- Jian Liu (ORCID: https://orcid.org/0000-0003-4363-5420)
- Z. Jiao
- Zehao Wei (ORCID: https://orcid.org/0009-0000-8993-670X)
Institutions
- Northwestern Polytechnical University (CN)
- Jinan City People's Hospital (CN)
- Shandong First Medical University (CN)
- First Hospital of Qinhuangdao (CN)
- Northeastern University (CN)
Publication Details
- Journal
- Photonics
- Published
- 2025-08-25
- DOI
- https://doi.org/10.3390/photonics12090850
- Primary Topic
- Optical Coherence Tomography Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China