Feature learning based on connectivity estimation for unbiased mammography mass classification

Guobin Li*, Reyer Zwiggelaar

*Awdur cyfatebol y gwaith hwn

Allbwn ymchwil: Cyfraniad at gyfnodolynErthygladolygiad gan gymheiriaid

16 Wedi eu Llwytho i Lawr (Pure)


Breast cancer is the most commonly diagnosed female malignancy worldwide. Recent developments in deep convolutional neural networks have shown promising performance for breast cancer detection and classification. However, due to variations in appearance and small datasets, biased features can be learned by the networks in distinguishing malignant and benign instances. To investigate these aspects, we trained a densely connected convolutional network (DenseNet) to obtain representative features of breast tissue, selecting texture features representing different physical morphological representations as the network's inputs. Connectivity estimation, represented by a connection matrix, is proposed for feature learning. To make the network provide an unbiased prediction, we used k-nearest neighbors to find k training samples whose connection matrices are closest to the test case. When evaluated on OMI-DB we achieved improved diagnostic accuracy 73.89±2.89% compared with 71.35±2.66% for the initial CNN model, which showed a statistically significant difference (p=0.00036). The k training samples can provide visual explanations which are useful in understanding the model predictions and failures of the model.

Iaith wreiddiolSaesneg
Rhif yr erthygl103884
Nifer y tudalennau10
CyfnodolynComputer Vision and Image Understanding
Dyddiad ar-lein cynnar14 Tach 2023
Dynodwyr Gwrthrych Digidol (DOIs)
StatwsCyhoeddwyd - 01 Ion 2024

Ôl bys

Gweld gwybodaeth am bynciau ymchwil 'Feature learning based on connectivity estimation for unbiased mammography mass classification'. Gyda’i gilydd, maen nhw’n ffurfio ôl bys unigryw.

Dyfynnu hyn