Mass Diagnosis in Mammography with Mutual Information Based Feature Selection and Support Vector Machine

Abstract

Mass classification is an important problem in breast cancer diagnosis. In this paper, we investigated the classification of masses with feature selection. Based on the initial contour guided by radiologist, level set algorithm is used to deform the contour and achieves the final segmentation. Morphological features are extracted from the boundary of segmented regions. Then, important features are extracted based on mutual information criterion. Linear discriminant analysis and support vector machine are investigated for the final classification. Mammography images from DDSM were used for experiment. The method achieved an accuracy of 86.6% with mutual information based feature selection and SVM classifier. The experimental result shows that mutual information based feature selection is useful for the diagnosis of masses.KeywordsMass diagnosisMammographyMutual informationfeature selectionSupport vector machine