Efficient Classification of Imbalanced Natural Disasters Data Using Generative Adversarial Networks for Data Augmentation

被引:8
作者
Eltehewy, Rokaya [1 ]
Abouelfarag, Ahmed [2 ]
Saleh, Sherine Nagy [3 ]
机构
[1] Arab Acad Sci & Technol AAST, Coll Engn & Technol, Comp Engn Dept, Cairo 2033, Egypt
[2] Arab Acad Sci & Technol AAST, Coll Artificial Intelligence, El Alamein 51718, Egypt
[3] Arab Acad Sci & Technol AAST, Coll Engn & Technol, Comp Engn Dept, Alexandria 1029, Egypt
关键词
data augmentation; deep neural network architectures; disaster classification; ensemble classifiers; generative adversarial networks;
D O I
10.3390/ijgi12060245
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Rapid damage identification and classification in disastrous situations and natural disasters are crucial for efficiently directing aid and resources. With the development of deep learning techniques and the availability of imagery content on social media platforms, extensive research has focused on damage assessment. Through the use of geospatial data related to such incidents, the visual characteristics of these images can quickly determine the safety situation in the region. However, training accurate disaster classification models has proven to be challenging due to the lack of labeled imagery data in this domain. This paper proposes a disaster classification framework, which combines a set of synthesized diverse disaster images generated using generative adversarial networks (GANs) and domain-specific fine-tuning of a deep convolutional neural network (CNN)-based model. The proposed model utilizes bootstrap aggregating (bagging) to further stabilize the target predictions. Since past work in this domain mainly suffers from limited data resources, a sample dataset that highlights the issue of imbalanced classification of multiple natural disasters was constructed and augmented. Qualitative and quantitative experiments show the validity of the data augmentation method employed in producing a balanced dataset. Further experiments with various evaluation metrics verified the proposed framework's accuracy and generalization ability across different classes for the task of disaster classification in comparison to other state-of-the-art techniques. Furthermore, the framework outperforms the other models by an average validation accuracy of 11%. These results provide a deep learning solution for real-time disaster monitoring systems to mitigate the loss of lives and properties.
引用
收藏
页数:18
相关论文
共 39 条
  • [31] Rethinking the Inception Architecture for Computer Vision
    Szegedy, Christian
    Vanhoucke, Vincent
    Ioffe, Sergey
    Shlens, Jon
    Wojna, Zbigniew
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 2818 - 2826
  • [32] Valdez D.B., 2021, P INT C ART INT ITS, P1, DOI [10.1145/3487923.3487927, DOI 10.1145/3487923.3487927]
  • [33] Generative Adversarial Networks in Computer Vision: A Survey and Taxonomy
    Wang, Zhengwei
    She, Qi
    Ward, Tomas E.
    [J]. ACM COMPUTING SURVEYS, 2021, 54 (02)
  • [34] Meta-Pose: Environment-adaptive Human Skeleton Tracking with RFID
    Yang, Chao
    Wang, Lingxiao
    Wang, Xuyu
    Mao, Shiwen
    [J]. 2021 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM), 2021,
  • [35] Transferability of Convolutional Neural Network Models for Identifying Damaged Buildings Due to Earthquake
    Yang, Wanting
    Zhang, Xianfeng
    Luo, Peng
    [J]. REMOTE SENSING, 2021, 13 (03)
  • [36] Yu Y., 2021, Frechet Inception Distance (Fid) for Evaluating Gans
  • [37] CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features
    Yun, Sangdoo
    Han, Dongyoon
    Oh, Seong Joon
    Chun, Sanghyuk
    Choe, Junsuk
    Yoo, Youngjoon
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 6022 - 6031
  • [38] Zahid Y, 2020, IEEE IMAGE PROC, P588, DOI 10.1109/ICIP40778.2020.9190673
  • [39] A Comprehensive Survey on Transfer Learning
    Zhuang, Fuzhen
    Qi, Zhiyuan
    Duan, Keyu
    Xi, Dongbo
    Zhu, Yongchun
    Zhu, Hengshu
    Xiong, Hui
    He, Qing
    [J]. PROCEEDINGS OF THE IEEE, 2021, 109 (01) : 43 - 76