arXiv:2501.03349cs.LGcs.AI2025-01被引 1

提出联邦迁移学习新方法,实现地质岩心图像高效分类且不泄露数据。

FTA-FTL: A Fine-Tuned Aggregation Federated Transfer Learning Scheme for Lithology Microscopic Image Classification

  • 基于微调聚合的联邦迁移学习框架,解决小样本与数据隐私难题。
  • 在多个指标上表现接近中心化训练,准确率接近95%。
  • 适合地质勘探、石油企业等需保护数据隐私的场景使用。

岩性识别是油藏表征的关键环节,处理岩心显微图像对化石、矿物研究及页岩油地质评估至关重要。深度学习可构建鲁棒分类模型,但大规模数据集难以获取。迁移学习与数据增强成为主流解决方案。由于数据隐私等因素,机构不愿共享敏感数据,联邦学习(FL)因此兴起,可在不传输原始数据的前提下联合训练高精度中心模型。本研究分为两阶段:第一阶段在小样本上通过迁移学习进行岩心图像分类,对比多种预训练模型架构;第二阶段将任务建模为联邦迁移学习(FTL),提出细调聚合策略(FTA-FTL)。实验采用准确率、F1分数、精确率、特异性、灵敏度(召回率)及混淆矩阵等多维度评估。结果表明,该方案性能优异,所提FTA-FTL算法在岩心图像分类任务中表现接近集中式实现,准确率可达约95%。

原文摘要 · Abstract (English)

Lithology discrimination is a crucial activity in characterizing oil reservoirs, and processing lithology microscopic images is an essential technique for investigating fossils and minerals and geological assessment of shale oil exploration. In this way, Deep Learning (DL) technique is a powerful approach for building robust classifier models. However, there is still a considerable challenge to collect and produce a large dataset. Transfer-learning and data augmentation techniques have emerged as popular approaches to tackle this problem. Furthermore, due to different reasons, especially data privacy, individuals, organizations, and industry companies often are not willing to share their sensitive data and information. Federated Learning (FL) has emerged to train a highly accurate central model across multiple decentralized edge servers without transferring sensitive data, preserving sensitive data, and enhancing security. This study involves two phases; the first phase is to conduct Lithology microscopic image classification on a small dataset using transfer learning. In doing so, various pre-trained DL model architectures are comprehensively compared for the classification task. In the second phase, we formulated the classification task to a Federated Transfer Learning (FTL) scheme and proposed a Fine-Tuned Aggregation strategy for Federated Learning (FTA-FTL). In order to perform a comprehensive experimental study, several metrics such as accuracy, f1 score, precision, specificity, sensitivity (recall), and confusion matrix are taken into account. The results are in excellent agreement and confirm the efficiency of the proposed scheme, and show that the proposed FTA-FTL algorithm is capable enough to achieve approximately the same results obtained by the centralized implementation for Lithology microscopic images classification task.

联邦学习岩性分类迁移学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。