LiTransUNet: a lightweight TransUNet for vertebral column segmentation
摘要
Precise segmentation is crucial for accurate computer-aided diagnosis systems, as segmenting the vertebral column from the medical images reduces noise and complexity, enabling more effective model training. However, identifying spinal discs and bones in medical images is challenging, especially when dealing with abnormalities. Although, Deep Convolutional Neural Networks (DCNNs) have shown potential in medical image segmentation, however, they often struggle with complex anatomical structures and data limitations, requiring significant computational resources, energy, and time. Current deep learning models for segmentation are typically highly parameterized, further increasing complexity. In response, this paper proposes a Lightweight deep learning framework, i.e., LiTransUNet, built on the top of TransUNet to segment Vertebral Columns from dog radiography samples. It uses depthwise separable and pointwise convolution along with a transformer to encode dataset samples instead of traditional CNNs. The proposed model is trained on radiographic images collected by the Department of Surgery and Radiology at the College of Veterinary Sciences, GBPUA&T, Pantnagar. Experimental results show the proposed method yields comparable performance while significantly reducing trainable parameters, memory usage, energy consumption, and processing time.