Jun, 2024

基于拓扑数据分析的 LightGBM 鲁棒优化算法

TL;DR提出了一种基于拓扑数据分析 (TDA) 的鲁棒性优化算法 TDA-LightGBM,用于增强光梯度提升机 (LightGBM) 图像分类算法的鲁棒性,以解决噪声对图像分类的干扰。实验证实了 TDA-LightGBM 在存在噪声情况下在五个分类任务上相对于 LightGBM 的准确性提升了 3%,在无噪声情况下在两个分类任务上相对于 LightGBM 的准确性提升了 0.5%,在乳腺癌超声乳房图像和遮挡的 CASIA WebFace 数据集上的分类准确性分别提高了 6% 和 15%,凸显了 TDA-LightGBM 方法在集成了拓扑特征的同时加强了 LightGBM 的鲁棒性,从而提高了在数据扰动中的图像分类任务的性能。