Both interpretation and accuracy are very important for a predictive model in real applications, but most of previous works, no matter interpretable models or black-box models, cannot simultaneously achieve both of them, resulting in a trade-off between model interpretation and model accuracy. To break this trade-off, in this paper, we propose a flexible framework, named IB-M, to align an Interpretable model and a Black-box Model for simultaneously optimizing model interpretation and model accuracy. Generally, we think most of samples that are well-clustered or away from the true decision boundary can be easily interpreted by an interpretable model. Removing those samples can help to learn a more accurate black-box model by focusing on the left samples around the true decision boundary. Inspired by this, we propose a data re-weighting based framework to align an interpretable model and a blackbox model, letting them focus on the samples what they are good at, hence, achieving both interpretation and accuracy. We implement our IB-M framework for a real medical problem of ultrasound thyroid nodule diagnosis. Extensive experiments demonstrate that our proposed framework and algorithm can achieve a more interpretable and more accurate diagnosis than a single interpretable model and a single black-box model.
基金:
National Natural Science Foundation of ChinaNational Natural Science Foundation of China (NSFC) [62006207, 62037001]; National Key Research and Development Program of China [2018AAA0101900, 2020YFC0832500]; Fundamental Research Funds for the Central UniversitiesFundamental Research Funds for the Central Universities
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]Zhejiang Univ, Hangzhou, Peoples R China
通讯作者:
推荐引用方式(GB/T 7714):
Li Mengze,Kuang Kun,Zhu Qiang,et al.IB-M: A Flexible Framework to Align an Interpretable Model and a Black-box Model[J].2020 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE.2020,643-649.doi:10.1109/BIBM49941.2020.9313119.
APA:
Li, Mengze,Kuang, Kun,Zhu, Qiang,Chen, Xiaohong,Guo, Qing&Wu, Fei.(2020).IB-M: A Flexible Framework to Align an Interpretable Model and a Black-box Model.2020 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE,,
MLA:
Li, Mengze,et al."IB-M: A Flexible Framework to Align an Interpretable Model and a Black-box Model".2020 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE .(2020):643-649