arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.01001cs.CVeess.IV

Attention Mechanism Meets with Hybrid Dense Network for Hyperspectral Image Classification

National University of Computer and Emerging Sciences(国立计算机与新兴科学大学);Innopolis University(因诺波利斯大学);University of Messina(墨西拿大学);Jalpaiguri Government Engineering College(贾尔派古里政府工程学院);Beijing Institute of Technology(北京理工大学)

Muhammad Ahmad, Adil Mehmood Khan, Manuel Mazzara, Salvatore Distefano, Swalpa Kumar Roy, Xin Wu

英文摘要:

Convolutional Neural Networks (CNN) are more suitable, indeed. However, fixed kernel sizes make traditional CNN too specific, neither flexible nor conducive to feature learning, thus impacting on the classification accuracy. The convolution of different kernel size networks may overcome this problem by capturing more discriminating and relevant information. In light of this, the proposed solution aims at combining the core idea of 3D and 2D Inception net with the Attention mechanism to boost the HSIC CNN performance in a hybrid scenario. The resulting \textit{attention-fused hybrid network} (AfNet) is based on three attention-fused parallel hybrid sub-nets with different kernels in each block repeatedly using high-level features to enhance the final ground-truth maps. In short, AfNet is able to selectively filter out the discriminative features critical for classification. Several tests on HSI datasets provided competitive results for AfNet compared to state-of-the-art models. The proposed pipeline achieved, indeed, an overall accuracy of 97\% for the Indian Pines, 100\% for Botswana, 99\% for Pavia University, Pavia Center, and Salinas datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑