Combination of loss functions for robust breast cancer prediction

被引:30
作者
Hajiabadi, Hamideh [1 ]
Babaiyan, Vahide [1 ]
Zabihzadeh, Davood [2 ]
Hajiabadi, Moein [1 ]
机构
[1] Birjand Univ Technol, Dept Comp Engn, Birjand, Iran
[2] Sabzevar Univ New Technol, Engn Fac, Comp Dept, Sabzevar, Iran
关键词
Artificial neural network; Breast cancer detection; Margin enhancing loss function; Robust loss function; ARTIFICIAL NEURAL-NETWORK; MACHINE; DIAGNOSIS;
D O I
10.1016/j.compeleceng.2020.106624
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Cancer detection can be formulated as a binary classification in a machine learning paradigm. Loss functions are a critical part of almost every machine learning algorithm. While each loss function comes up with its own advantages and disadvantages, in this paper, inspired by ensemble methods, we propose a novel objective function that is a linear combination of single losses. We then integrate the proposed objective function into an Artificial Neural Network (ANN) to diagnose breast cancer. By doing so, both the coefficients of loss functions and weight parameters of the ANN are learned jointly via backpropagation. As the patients' data are sometimes very noisy, we evaluate our method by doing comprehensive experiments on Wisconsin Breast Cancer Diagnosis (WBCD) dataset at different noise levels. The experiments show its performance declines very slowly (from 0.97 to 0.96) compared to the peer methods with the increase of noise level. (C) 2020 Elsevier Ltd. All rights reserved.
引用
收藏
页数:12
相关论文
共 30 条
[1]   Using machine learning tool in classification of breast cancer [J].
Abdel-Ilah, Layla ;
Sahinbegovic, Hana .
PROCEEDINGS OF THE INTERNATIONAL CONFERENCE ON MEDICAL AND BIOLOGICAL ENGINEERING 2017 (CMBEBIH 2017), 2017, 62 :3-8
[2]  
[Anonymous], 2011, P S MACH LEARN SPEEC
[3]  
Aslan M.F., 2018, International Journal of Intelligent Systems and Applications in Engineering, V6, P289, DOI [10.18201/ijisae.2018648455, DOI 10.18201/IJISAE.2018648455]
[4]   Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis [J].
Asri, Hiba ;
Mousannif, Hajar ;
Al Moatassime, Hassan ;
Noel, Thomas .
7TH INTERNATIONAL CONFERENCE ON AMBIENT SYSTEMS, NETWORKS AND TECHNOLOGIES (ANT 2016) / THE 6TH INTERNATIONAL CONFERENCE ON SUSTAINABLE ENERGY INFORMATION TECHNOLOGY (SEIT-2016) / AFFILIATED WORKSHOPS, 2016, 83 :1064-1069
[5]   Breast cancer diagnosis using Genetically Optimized Neural Network model [J].
Bhardwaj, Arpit ;
Tiwari, Aruna .
EXPERT SYSTEMS WITH APPLICATIONS, 2015, 42 (10) :4611-4620
[6]   Deep learning beyond cats and dogs: recent advances in diagnosing breast cancer with deep neural networks [J].
Burt, Jeremy R. ;
Torosdagli, Neslisah ;
Khosravan, Naji ;
Raviprakash, Harish ;
Mortazi, Aliasghar ;
Tissavirasingham, Fiona ;
Hussein, Sarfaraz ;
Bagci, Ulas .
BRITISH JOURNAL OF RADIOLOGY, 2018, 91 (1089)
[7]   Prediction of scour depth around bridge piers using self-adaptive extreme learning machine [J].
Ebtehaj, Isa ;
Sattar, Ahmed M. A. ;
Bonakdari, Hossein ;
Zaji, Amir Hossein .
JOURNAL OF HYDROINFORMATICS, 2017, 19 (02) :207-224
[8]  
Hajiabadi H, 2017, P AUSTR LANG TECHN A
[9]  
Hajiabadi H., ARXIV1810110712018
[10]   Layered Geometric Learning [J].
Hajiabadi, Hamideh ;
Godaz, Reza ;
Ghasemi, Morteza ;
Monsefi, Reza .
ARTIFICIAL INTELLIGENCEAND SOFT COMPUTING, PT I, 2019, 11508 :571-582