Dual-View Pyramid Pooling in Deep Neural Networks for Improved Medical Image Classification and Confidence Calibration

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Xiaoqing, Nie, Qiushi, Xiao, Zunjie, Zhao, Jilu, Wu, Xiao, Guo, Pengxin, Li, Runzhi, Liu, Jin, Wei, Yanjie, Pan, Yi
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916358188433408
author Zhang, Xiaoqing
Nie, Qiushi
Xiao, Zunjie
Zhao, Jilu
Wu, Xiao
Guo, Pengxin
Li, Runzhi
Liu, Jin
Wei, Yanjie
Pan, Yi
author_facet Zhang, Xiaoqing
Nie, Qiushi
Xiao, Zunjie
Zhao, Jilu
Wu, Xiao
Guo, Pengxin
Li, Runzhi
Liu, Jin
Wei, Yanjie
Pan, Yi
contents Spatial pooling (SP) and cross-channel pooling (CCP) operators have been applied to aggregate spatial features and pixel-wise features from feature maps in deep neural networks (DNNs), respectively. Their main goal is to reduce computation and memory overhead without visibly weakening the performance of DNNs. However, SP often faces the problem of losing the subtle feature representations, while CCP has a high possibility of ignoring salient feature representations, which may lead to both miscalibration of confidence issues and suboptimal medical classification results. To address these problems, we propose a novel dual-view framework, the first to systematically investigate the relative roles of SP and CCP by analyzing the difference between spatial features and pixel-wise features. Based on this framework, we propose a new pooling method, termed dual-view pyramid pooling (DVPP), to aggregate multi-scale dual-view features. DVPP aims to boost both medical image classification and confidence calibration performance by fully leveraging the merits of SP and CCP operators from a dual-axis perspective. Additionally, we discuss how to fulfill DVPP with five parameter-free implementations. Extensive experiments on six 2D/3D medical image classification tasks show that our DVPP surpasses state-of-the-art pooling methods in terms of medical image classification results and confidence calibration across different DNNs.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02906
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dual-View Pyramid Pooling in Deep Neural Networks for Improved Medical Image Classification and Confidence Calibration
Zhang, Xiaoqing
Nie, Qiushi
Xiao, Zunjie
Zhao, Jilu
Wu, Xiao
Guo, Pengxin
Li, Runzhi
Liu, Jin
Wei, Yanjie
Pan, Yi
Computer Vision and Pattern Recognition
Spatial pooling (SP) and cross-channel pooling (CCP) operators have been applied to aggregate spatial features and pixel-wise features from feature maps in deep neural networks (DNNs), respectively. Their main goal is to reduce computation and memory overhead without visibly weakening the performance of DNNs. However, SP often faces the problem of losing the subtle feature representations, while CCP has a high possibility of ignoring salient feature representations, which may lead to both miscalibration of confidence issues and suboptimal medical classification results. To address these problems, we propose a novel dual-view framework, the first to systematically investigate the relative roles of SP and CCP by analyzing the difference between spatial features and pixel-wise features. Based on this framework, we propose a new pooling method, termed dual-view pyramid pooling (DVPP), to aggregate multi-scale dual-view features. DVPP aims to boost both medical image classification and confidence calibration performance by fully leveraging the merits of SP and CCP operators from a dual-axis perspective. Additionally, we discuss how to fulfill DVPP with five parameter-free implementations. Extensive experiments on six 2D/3D medical image classification tasks show that our DVPP surpasses state-of-the-art pooling methods in terms of medical image classification results and confidence calibration across different DNNs.
title Dual-View Pyramid Pooling in Deep Neural Networks for Improved Medical Image Classification and Confidence Calibration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.02906