Deep structure learning using feature extraction in trained projection space

被引:3
作者
Angermann, Christoph [1 ]
Haltmeier, Markus [1 ]
机构
[1] Univ Innsbruck, Dept Math, Technikerstr 13, A-6020 Innsbruck, Austria
关键词
Projection network; Learned Radon-transform; Volumetric segmentation; Memory efficiency; Implicit data augmentation; X-ray transform; Learned fusion;
D O I
10.1016/j.compeleceng.2021.107097
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Over the last decade of machine learning, convolutional neural networks have been the most striking successes for feature extraction of rich sensory and high-dimensional data. While learning data representations via convolutions is already well studied and efficiently implemented in various deep learning libraries, one often faces limited memory capacity and insufficient number of training data, especially for high-dimensional and large-scale tasks. To overcome these limitations, we introduce a network architecture using a self-adjusting and data dependent version of the Radon-transform (linear data projection), also known as Xray projection, to enable feature extraction via convolutions in lower-dimensional space. The resulting framework, named PiNet, can be trained end-to-end and shows promising performance on volumetric segmentation tasks. We test proposed model on public datasets to show that our approach achieves comparable results only using fractional amount of parameters. Investigation of memory usage and processing time confirms PiNet's superior efficiency compared to other segmentation models.
引用
收藏
页数:8
相关论文
共 22 条
[1]   Projection-Based 2.5D U-net Architecture for Fast Volumetric Segmentation [J].
Angermann, Christoph ;
Haltmeier, Markus ;
Steiger, Ruth ;
Pereverzyev, Sergiy, Jr. ;
Gizewski, Elke .
2019 13TH INTERNATIONAL CONFERENCE ON SAMPLING THEORY AND APPLICATIONS (SAMPTA), 2019,
[2]  
[Anonymous], 2007, The Radon Transform and Some of Its Applications
[3]  
[Anonymous], 2012, ADADELTA ADAPTIVE LE
[4]  
Bishop M.C., 1995, Neural Networks for Pattern Recognition
[5]  
Chollet F., 2015, Keras
[6]  
Cicek Ozgun, 2016, Medical Image Computing and Computer-Assisted Intervention - MICCAI 2016. 19th International Conference. Proceedings: LNCS 9901, P424, DOI 10.1007/978-3-319-46723-8_49
[7]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[8]   Optimization for Medical Image Segmentation: Theory and Practice When Evaluating With Dice Score or Jaccard Index [J].
Eelbode, Tom ;
Bertels, Jeroen ;
Berman, Maxim ;
Vandermeulen, Dirk ;
Maes, Frederik ;
Bisschops, Raf ;
Blaschko, Matthew B. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (11) :3679-3690
[9]  
Eigen D, 2014, ADV NEUR IN, V27
[10]   3D U-Net for Skull Stripping in Brain MRI [J].
Hwang, Hyunho ;
Rehman, Hafiz Zia Ur ;
Lee, Sungon .
APPLIED SCIENCES-BASEL, 2019, 9 (03)