对抗制
生成语法
机制(生物学)
计算机科学
断层(地质)
生成对抗网络
人工智能
地质学
物理
深度学习
量子力学
地震学
作者
Tongfei Lei,Zeyu Pei,Feng Pan,Bing Li,Yuan Xu,Haidong Shao,Ke Zhao
标识
DOI:10.1088/1361-6501/ad2969
摘要
Abstract Learning the original data distribution and generating new samples has proven to be an effective approach in addressing the issue of data imbalance. This paper combines the strengths of generative adversarial networks and variational autoencoder, proposing a novel data augmentation method named variational autoencoding generative adversarial networks with self-attention. Specifically, an encoding-decoding process is introduced during the generative adversarial process to provide distribution information for the generator’s sampling space, thereby accelerating the model’s convergence speed and simultaneously improving the quality of generated samples. Additionally, a self-attention module is incorporated into the discriminator to capture global information from the input data, guiding the generator. During the training process, overlapping sampling and feature-layer matching are employed. Comparative experiments with other advanced algorithms on both public and engineering datasets with multiple imbalanced cases demonstrate that the proposed method can generate high-quality samples, effectively enhance original imbalanced data, and exhibit strong generalization and robustness.
科研通智能强力驱动
Strongly Powered by AbleSci AI