轻量化框架StrokeSeg2让脑卒中病灶分割模型轻松部署到临床工作站。
StrokeSeg2: Stroke Lesion Segmentation in Clinical Research Workflows

- 用知识蒸馏和量化压缩模型,大幅降低资源消耗。
- 模型缩小至2.1MB,推理时间减少84%,能耗降90%以上。
- 无需依赖Docker,支持多平台一键安装,适合临床研究使用。
深度学习框架如nnU-Net虽在脑卒中病灶分割上表现优异,但因软件依赖和计算需求高,难以在临床研究环境中部署。本文提出StrokeSeg2,一个轻量级、模块化、跨平台的C++/Qt框架,可将资源密集型3D卒中分割流程转化为便携、可复现的应用程序。为提升与标准临床工作站的兼容性,研究了通过知识蒸馏进行架构压缩,以及使用ONNX Runtime结合Float16量化进行推理优化的联合效果。在异构硬件(CPU、集成显卡、独立显卡)上,架构压缩成为效率提升的主要因素,贡献了超过90%的能耗降低和平均84%的推理时间减少。特别地,识别出参数量仅0.84M的轻量学生模型为最优平衡点,将原102.3M参数教师模型压缩至2.1MB磁盘占用,同时保持良好的病灶定位与竞争性分割性能。该小体积支持开发针对临床工作站的自包含安装包。最终,StrokeSeg2将这些优化打包为Windows、macOS和Linux的独立安装程序,提供图形界面与命令行接口,无需Docker或外部环境依赖,助力高性能分割工作流在常规临床研究流程中的部署。
原文摘要 · Abstract (English)
Deep learning frameworks like nnU-Net achieve state-of-theart brain lesion segmentation performance but remain difficult to deploy in clinical research environments due to, among other reasons, software dependencies and computational requirements. We introduce StrokeSeg2, a lightweight, modular, cross-platform C++/Qt framework designed to adapt resource-intensive 3D stroke segmentation pipelines into portable and reproducible applications. To improve compatibility with standard clinical workstations, we investigate the combined effect of architectural compression through knowledge distillation and inference optimisation using ONNX Runtime with Float16 quantisation. Across heterogeneous hardware configurations (CPU, integrated GPU, and dedicated GPU) architectural distillation emerged as the primary contributor to efficiency gains, contributing to over 90% reduction in energy consumption and an average 84% reduction in inference time. Specifically, we identify a 0.84M-parameter student model as the most favourable trade-off, reducing the original 102.3M-parameter teacher architecture to a 2.1 MB disk footprint while preserving robust lesion localisation and competitive segmentation performance. This small footprint supports the development of a self-contained installer for clinical workstation targets. Finally, StrokeSeg2 packages these optimisations into standalone installers for Windows, macOS, and Linux. By providing both graphical and commandline interfaces without Docker or external environment dependencies, StrokeSeg2 facilitates deployment of high-performance segmentation workflows for routine clinical research pipelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。