Title :
An investigate of mass diagnosis in mammogram with random forest
Author :
Liu, Jun ; Chen, Jianxun ; Liu, Xiaoming ; Tang, J.
Author_Institution :
Coll. of Comput. Sci. & Technol., Wuhan Univ. of Sci. & Technol., Wuhan, China
Abstract :
Correct mass diagnosis in mammogram can reduce the unnecessary biopsy without increasing false negatives. In this paper, we investigated the usage of random forest classifier for the classification of masses with geometry and texture features. Before extracting features, the mass regions need to be extracted. Based on the initial contour guided by radiologist, level set segmentation is used to deform the contour and achieves the final segmentation. The proposed level set method integrated both region information and boundary information, and with a level regularization term, it can achieve accurate segmentation. Modified Hu moments were used for shape characteristics, and GLCM (Gray Level Co-occurrence Matrix) features are used for texture characteristics. Random forest, a recently proposed ensemble learning method, for the first time, is investigated for the mass classification, and is compared with SVM (Support Vector Machine). Mammography images from DDSM were used for experiment. The new method based on the level set segmentation and the features achieved a Az value of 0.86 with SVM and 0.83 with random forest. The experimental result shows that random forest is a promising method for the diagnosis of masses.
Keywords :
feature extraction; image classification; image segmentation; image texture; learning (artificial intelligence); mammography; matrix algebra; medical image processing; patient diagnosis; set theory; support vector machines; Hu moments; SVM; biopsy; boundary information; ensemble learning method; feature extraction; geometry feature; gray level cooccurrence matrix; level regularization term; level set segmentation; mammogram; mass diagnosis; masses classification; radiologist; random forest classifier; region information; shape characteristics; support vector machine; texture feature; Accuracy; Cancer; Feature extraction; Image segmentation; Level set; Radio frequency; Support vector machines;
Conference_Titel :
Advanced Computational Intelligence (IWACI), 2011 Fourth International Workshop on
Conference_Location :
Wuhan
Print_ISBN :
978-1-61284-374-2
DOI :
10.1109/IWACI.2011.6160086