ORALI: Open-set recognition and active learning for unknown lithology identification

Xinyi Zhu,Hongbing Zhang,Quan Ren,Jianwen Rui,Lingyuan Zhang,Dailu Zhang
DOI: https://doi.org/10.1016/j.engappai.2024.108623
IF: 8
2024-05-16
Engineering Applications of Artificial Intelligence
Abstract:Lithology identification using logging data is an essential part of geophysical reservoir characterization. Reliable core samples are very limited, which means that it is usually infeasible to obtain all the desired lithology categories. Unknown lithologies can arise in testing, while traditional closed-set models are not applicable to such open environments. Therefore, this paper proposes a framework based on open-set recognition and active learning for lithology identification (ORALI). This framework is two-stage. Firstly, a rejection measure is added to the open-set nearest neighbor (OSNN), which considers the class ratios within sub-clusters of Gaussian mixture models (GMM). This can effectively improve the identification of unknowns in areas of class overlap. After the unknown lithologies are identified, expert labelling is performed using active learning. A cold-start batch-mode processing strategy is designed to efficiently perform queries with unknown lithologies. In the comparison experiments of actual logging lithology identification, the performance of ORALI is excellent and most balanced. In both study areas in the Pearl River Mouth Basin, ORALI has achieved recognition accuracies of over 88% in the presence of unknown lithologies. Among them, the F1 scores for unknown lithology recognition reached 81.73% and 82.67% respectively, which were improved by 7.8% and 8.2% compared with OSNN. Meanwhile, ORALI can initiate and query more effectively in the active learning stage. ORALI provides a complete and effective framework for open-set lithology identification task.
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary
What problem does this paper attempt to address?