Saved in:
Bibliographic Details
Main Authors: RahimiZadeh, Keyvan, Taheri, Ahmad, Baumbach, Jan, Makarian, Esmael, Dehghani, Abbas, Ravaei, Bahman, Javadi, Bahman, Beheshti, Amin
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2501.03349
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909449608757248
author RahimiZadeh, Keyvan
Taheri, Ahmad
Baumbach, Jan
Makarian, Esmael
Dehghani, Abbas
Ravaei, Bahman
Javadi, Bahman
Beheshti, Amin
author_facet RahimiZadeh, Keyvan
Taheri, Ahmad
Baumbach, Jan
Makarian, Esmael
Dehghani, Abbas
Ravaei, Bahman
Javadi, Bahman
Beheshti, Amin
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03349
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FTA-FTL: A Fine-Tuned Aggregation Federated Transfer Learning Scheme for Lithology Microscopic Image Classification
RahimiZadeh, Keyvan
Taheri, Ahmad
Baumbach, Jan
Makarian, Esmael
Dehghani, Abbas
Ravaei, Bahman
Javadi, Bahman
Beheshti, Amin
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
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.
title FTA-FTL: A Fine-Tuned Aggregation Federated Transfer Learning Scheme for Lithology Microscopic Image Classification
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2501.03349