English

Sculpting Efficiency: Pruning Medical Imaging Models for On-Device Inference

Computer Vision and Pattern Recognition 2023-11-03 v2

Abstract

Leveraging ML advancements to augment healthcare systems can improve patient outcomes. Yet, uninformed engineering decisions in early-stage research inadvertently hinder the feasibility of such solutions for high-throughput, on-device inference, particularly in settings involving legacy hardware and multi-modal gigapixel images. Through a preliminary case study concerning segmentation in cardiology, we highlight the excess operational complexity in a suboptimally configured ML model from prior work and demonstrate that it can be sculpted away using pruning to meet deployment criteria. Our results show a compression rate of 1148x with minimal loss in quality (~4%) and, at higher rates, achieve faster inference on a CPU than the GPU baseline, stressing the need to consider task complexity and architectural details when using off-the-shelf models. With this, we consider avenues for future research in streamlining workflows for clinical researchers to develop models quicker and better suited for real-world use.

Keywords

Cite

@article{arxiv.2309.05090,
  title  = {Sculpting Efficiency: Pruning Medical Imaging Models for On-Device Inference},
  author = {Sudarshan Sreeram and Bernhard Kainz},
  journal= {arXiv preprint arXiv:2309.05090},
  year   = {2023}
}

Comments

Accepted at MedNeurIPS 2023

R2 v1 2026-06-28T12:17:27.414Z