Breast mass segmentation based on ultrasonic entropy maps and attention gated U-Net
Authors:
Michal Byra,
Piotr Jarosik,
Katarzyna Dobruch-Sobczak,
Ziemowit Klimonda,
Hanna Piotrzkowska-Wroblewska,
Jerzy Litniewski,
Andrzej Nowicki
Abstract:
We propose a novel deep learning based approach to breast mass segmentation in ultrasound (US) imaging. In comparison to commonly applied segmentation methods, which use US images, our approach is based on quantitative entropy parametric maps. To segment the breast masses we utilized an attention gated U-Net convolutional neural network. US images and entropy maps were generated based on raw US si…
▽ More
We propose a novel deep learning based approach to breast mass segmentation in ultrasound (US) imaging. In comparison to commonly applied segmentation methods, which use US images, our approach is based on quantitative entropy parametric maps. To segment the breast masses we utilized an attention gated U-Net convolutional neural network. US images and entropy maps were generated based on raw US signals collected from 269 breast masses. The segmentation networks were developed separately using US image and entropy maps, and evaluated on a test set of 81 breast masses. The attention U-Net trained based on entropy maps achieved average Dice score of 0.60 (median 0.71), while for the model trained using US images we obtained average Dice score of 0.53 (median 0.59). Our work presents the feasibility of using quantitative US parametric maps for the breast mass segmentation. The obtained results suggest that US parametric maps, which provide the information about local tissue scattering properties, might be more suitable for the development of breast mass segmentation methods than regular US images.
△ Less
Submitted 27 January, 2020;
originally announced January 2020.
WaveFlow - Towards Integration of Ultrasound Processing with Deep Learning
Authors:
Piotr Jarosik,
Michał Byra,
Marcin Lewandowski
Abstract:
The ultimate goal of this work is a real-time processing framework for ultrasound image reconstruction augmented with machine learning. To attain this, we have implemented WaveFlow - a set of ultrasound data acquisition and processing tools for TensorFlow. WaveFlow includes: ultrasound Environments (connection points between the input raw ultrasound data source and TensorFlow) and signal processin…
▽ More
The ultimate goal of this work is a real-time processing framework for ultrasound image reconstruction augmented with machine learning. To attain this, we have implemented WaveFlow - a set of ultrasound data acquisition and processing tools for TensorFlow. WaveFlow includes: ultrasound Environments (connection points between the input raw ultrasound data source and TensorFlow) and signal processing Operators (ops) library. Raw data can be processed in real-time using algorithms available both in TensorFlow and WaveFlow. Currently, WaveFlow provides ops for B-mode image reconstruction (beamforming), signal processing and quantitative ultrasound. The ops were implemented both for the CPU and GPU, as well as for built-in automated tests and benchmarks. To demonstrate WaveFlow's performance, ultrasound data were acquired from wire and cyst phantoms and elaborated using selected sequences of the ops. We implemented and evaluated: Delay-and-Sum beamformer, synthetic transmit aperture imaging (STAI), plane-wave imaging (PWI), envelope detection algorithm and dynamic range clipping. The benchmarks were executed on the NVidia Titan X GPU integrated in the USPlatform research scanner (us4us Ltd., Poland). We achieved B-mode image reconstruction frame rates of 55 fps, 17 fps for the STAI and the PWI algorithms, respectively. The results showed the feasibility of real-time ultrasound image reconstruction using WaveFlow operators in the TensorFlow framework. WaveFlow source code can be found at github.com/waveflow-team/waveflow
△ Less
Submitted 5 November, 2018;
originally announced November 2018.
