Suppr超能文献

BSDR:一种基于离散松弛的高效数据深度学习高光谱波段选择算法

BSDR: A Data-Efficient Deep Learning-Based Hyperspectral Band Selection Algorithm Using Discrete Relaxation.

作者信息

Rahman Mohammad, Teng Shyh Wei, Murshed Manzur, Paul Manoranjan, Brennan David

机构信息

Institute of Innovation, Science and Sustainability, Federation University Australia, University Drive, Mt Helen, VIC 3350, Australia.

Cooperative Research Centre for High Performance Soils, Callaghan, NSW 2308, Australia.

出版信息

Sensors (Basel). 2024 Dec 4;24(23):7771. doi: 10.3390/s24237771.

Abstract

Hyperspectral band selection algorithms are crucial for processing high-dimensional data, which enables dimensionality reduction, improves data analysis, and enhances computational efficiency. Among these, attention-based algorithms have gained prominence by ranking bands based on their discriminative capability. However, they require a large number of model parameters, which increases the need for extensive training data. To address this challenge, we propose Band Selection through Discrete Relaxation (BSDR), a novel deep learning-based algorithm. BSDR reduces the number of learnable parameters by focusing solely on the target bands, which are typically far fewer than the original bands, thus resulting in a data-efficient configuration that minimizes training data requirements and reduces training time. The algorithm employs discrete relaxation, transforming the discrete problem of band selection into a continuous optimization task, which enables gradient-based search across the spectral dimension. Through extensive evaluations on three benchmark datasets with varying spectral dimensions and characteristics, BSDR demonstrates superior performance for both regression and classification tasks, achieving up to 25% and 34.6% improvements in overall accuracy, compared to the latest attention-based and traditional algorithms, respectively, while reducing execution time by 96.8% and 97.18%. These findings highlight BSDR's effectiveness in addressing key challenges in hyperspectral band selection.

摘要

高光谱波段选择算法对于处理高维数据至关重要,它能够实现降维、改善数据分析并提高计算效率。其中,基于注意力的算法通过根据波段的判别能力对波段进行排序而备受关注。然而,它们需要大量的模型参数,这增加了对大量训练数据的需求。为应对这一挑战,我们提出了通过离散松弛进行波段选择(BSDR),这是一种基于深度学习的新型算法。BSDR通过仅关注目标波段来减少可学习参数的数量,目标波段通常比原始波段少得多,从而形成一种数据高效的配置,将训练数据需求降至最低并减少训练时间。该算法采用离散松弛,将波段选择的离散问题转化为连续优化任务,从而能够在光谱维度上进行基于梯度的搜索。通过对三个具有不同光谱维度和特征的基准数据集进行广泛评估,BSDR在回归和分类任务中均表现出卓越性能,与最新的基于注意力的算法和传统算法相比,总体准确率分别提高了25%和34.6%,同时执行时间分别减少了96.8%和97.18%。这些发现凸显了BSDR在应对高光谱波段选择关键挑战方面的有效性。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/0963/11644938/fa122ce97ceb/sensors-24-07771-g001.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验