Gupta V, Bhavsar A. Heterogeneous ensemble with information theoretic diversity measure for human epithelial cell image classification.
Med Biol Eng Comput 2021;
59:1035-1054. [PMID:
33860445 DOI:
10.1007/s11517-021-02336-8]
[Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 11/14/2019] [Accepted: 02/09/2021] [Indexed: 12/01/2022]
Abstract
In this work, we propose a heterogeneous committee (ensemble) of diverse members (classification approaches) to solve the problem of human epithelial (HEp-2) cell image classification using indirect Immunofluorescence (IIF) imaging. We hypothesize that an ensemble involving different feature representations can enable higher performance if individual members in the ensemble are sufficiently varied. These members are of two types: (1) CNN-based members, (2) traditional members. For the CNN members, we have employed the well-established ResNet, DenseNet, and Inception models, which have distinctive salient aspects. For the traditional members, we incorporate class-specific features which are characterized depending on visual morphological attributes, and some standard texture features. To select the members which are discriminating and not redundant, we use an information theoretic measure which considers the trade-off between individual accuracies and diversity among the members. For all selected members, a compelling fusion required to combine their outputs to reach a final decision. Thus, we also investigate various fusion methods that combine the opinion of the committee at different levels: maximum voting, product, decision template, Bayes, Dempster-Shafer, etc. The proposed method is evaluated using ICPR-2014 data which consists of more images than some previous datasets ICPR-2012 and demonstrate state-of-the-art performance. To check the effectiveness of the proposed methodology for other related datasets, we test our methodology with newly compiled large-scale HEp-2 dataset with 63K cell images and demonstrate comparable performance even with less number of training samples. The proposed method produces 99.80% and 86.03% accuracy respectively when tested on ICPR-2014 and a new large-scale data containing 63K samples. Graphical Abstract Overview of the proposed methodology.
Collapse