Dual-input attention network for automatic identification of detritus from river sands

• Published in: Computers & geosciences 2021-06, Vol.151, p.104735, Article 104735
• Main Authors: Ge, Shiping, Wang, Cong, Jiang, Zhiwei, Hao, Huizhen, Gu, Qing
• Format: Article
• Language: English
• Subjects: Attention mechanism; Convolutional neural network; Detritus Identification; Image classification; River sands
• ISSN: 0098-3004; 1873-7803
• DOI: 10.1016/j.cageo.2021.104735

Identifying the categories of detritus collected from river sands is an important work in geological researches, including sediment source analysis, tectonic evolution and lithofacies palaeogeography. Among deep learning techniques developed in recent years, Convolutional Neural Network (CNN) can be applied to the detritus identification problem. However, due to both data insufficiency caused by the high cost of manual labelling, and data imbalance caused by the uneven distribution of different categories of detritus, existing CNN models are hindered to reach their best performance. In this paper, we propose a novel network architecture for the problem of detritus identification: Dual-Input Attention Network (DANet), which accepts both plane-polarized images and cross-polarized images of detritus as input, and uses Parametrized Cross-Entropy as the loss function in order to alleviate the poor performance of detritus identification caused by data insufficiency and data imbalance. Experiments based on the detritus collected from the Yarlung Zangbo River Basin prove both the effectiveness and potential of DANet for detritus identification. •A novel network architecture named DANet is proposed for the identification of detritus from river sands.•DANet accepts both plane-polarized images and cross-polarized images as input.•Parametrized Cross-Entropy is proposed as the loss function of DANet.•Experiments prove both the effectiveness and potential of DANet for detritus identification.