CAFlow通过自适应深度路由实现高效病理图像超分辨,显著降低计算量。
CAFlow: Adaptive-Depth Single-Step Flow Matching for Efficient Histopathology Super-Resolution
- 根据重建质量动态选择最浅网络出口,减少计算开销。
- 在多器官组织图像上达到31.72 dB PSNR,比全深度少用2.8倍算力。
- 支持单卡5小时内训练,全片推理从分钟级降至秒级,适合临床部署。
数字病理中全片图像常超过吉像素分辨率,导致生成式超分辨(SR)计算成本过高。本文提出CAFlow,一种自适应深度的单步流匹配框架,将每张图像块路由至能保持重建质量的最浅网络出口。该方法在未打乱像素的空间中进行流匹配,使空间计算减少16倍,并支持直接推理。实验表明,训练时投入一半数据于t=0精确样本对单步质量至关重要(不使用则下降1.5 dB)。主干网络FlowResNet(190万参数)在四个早期出口间混合卷积与窗口自注意力模块,算力范围为3.1至13.3 GFLOPs。轻量级出口分类器(约6千参数)实现33%算力节省,仅损失0.12 dB。在多器官组织图像x4超分辨任务中,自适应路由达31.72 dB PSNR,较全深度低0.12 dB,但算力仅为SwinIR-light的2.8倍;在最浅出口下仍优于双三次插值1.9 dB。模型在未见结肠组织上表现稳定(仅降0.02 dB),在x8超分时超越所有同规模基线,且媲美更大规模的SwinIR-Medium。下游核分割验证了关键结构得以保留。模型可在单张GPU上5小时内完成训练,自适应路由使全片推理时间从分钟级压缩至秒级。
原文摘要 · Abstract (English)
In digital pathology, whole-slide images routinely exceed gigapixel resolution, making computationally intensive generative super-resolution (SR) impractical for routine deployment. We introduce CAFlow, an adaptive-depth single-step flow-matching framework that routes each image tile to the shallowest network exit that preserves reconstruction quality. CAFlow performs flow matching in pixel-unshuffled rearranged space, reducing spatial computation by 16x while enabling direct inference. We show that dedicating half of training to exact t=0 samples is essential for single-step quality (-1.5 dB without it). The backbone, FlowResNet (1.90M parameters), mixes convolution and window self-attention blocks across four early exits spanning 3.1 to 13.3 GFLOPs. A lightweight exit classifier (~6K parameters) achieves 33% compute savings at only 0.12 dB cost. On multi-organ histopathology x4 SR, adaptive routing achieves 31.72 dB PSNR versus 31.84 dB at full depth, while the shallowest exit exceeds bicubic by +1.9 dB at 2.8x less compute than SwinIR-light. The method generalizes to held-out colon tissue with minimal quality loss (-0.02 dB), and at x8 upscaling it outperforms all comparable-compute baselines while remaining competitive with the much larger SwinIR-Medium model. Downstream nuclei segmentation confirms preservation of clinically relevant structure. The model trains in under 5 hours on a single GPU, and adaptive routing can reduce whole-slide inference from minutes to seconds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。