A Cross-Attention Multi-Scale Performer With Gaussian Bit-Flips for File Fragment Classification

被引:0
|
作者
Liu, Sisung [1 ]
Park, Jeong Gyu [2 ]
Kim, Hyeongsik [3 ]
Hong, Je Hyeong [1 ,3 ]
机构
[1] Hanyang Univ, Dept Artificial Intelligence, Seoul 04763, South Korea
[2] Hanyang Univ, Dept Elect Engn, Seoul 04763, South Korea
[3] Hanyang Univ, Dept Artificial Intelligence Semicond Engn, Seoul 04763, South Korea
关键词
Transformers; Feature extraction; Data models; Adaptation models; Accuracy; Attention mechanisms; Computational modeling; Training; Electronic mail; Data augmentation; File fragment classification; transformer; multi-scale attention; cross-attention; performer;
D O I
10.1109/TIFS.2025.3539527
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
File fragment classification is a crucial task in digital forensics and cybersecurity, and has recently achieved significant improvement through the deployment of convolutional neural networks (CNNs) compared to traditional handcrafted feature-based methods. However, CNN-based models exhibit inherent biases that can limit their effectiveness for larger datasets. To address this limitation, we propose the Cross-Attention Multi-Scale Performer (XMP) model, which integrates the attention mechanisms of transformer encoders with the feature extraction capabilities of CNNs. Compared to our conference work, we additionally introduce a new Gaussian Bit-Flip (GBFlip) method for binary data augmentation, largely inspired by bit flipping errors in digital system, improving the model performance. Furthermore, we incorporate a fine-tuning approach and demonstrate XMP adapts more effectively to diverse datasets than other CNN-based competitors without extensive hyperparameter tuning. Our experimental results on two public file fragment classification datasets show XMP surpassing other CNN-based and RCNN-based models, achieving state-of-the-art performance in file fragment classification both with and without fine-tuning. Our code is available at https://github.com/DominicoRyu/XMP_TIFS.
引用
收藏
页码:2109 / 2121
页数:13
相关论文
共 50 条
  • [41] Deep Multi-Scale Representation Learning with Attention for Automatic Modulation Classification
    Wu, Xiaowei
    Wei, Shengyun
    Zhou, Yan
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [42] A temporal multi-scale hybrid attention network for sleep stage classification
    Jin, Zheng
    Jia, Kebin
    MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2023, 61 (09) : 2291 - 2303
  • [43] Densely Connected CNN with Multi-scale Feature Attention for Text Classification
    Wang, Shiyao
    Huang, Minlie
    Deng, Zhidong
    PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2018, : 4468 - 4474
  • [44] MSLANet: multi-scale long attention network for skin lesion classification
    Wan, Yecong
    Cheng, Yuanshuo
    Shao, Mingwen
    APPLIED INTELLIGENCE, 2023, 53 (10) : 12580 - 12598
  • [45] HYPERSPECTRAL IMAGE CLASSIFICATION VIA MULTI-SCALE RESIDUAL ATTENTION NETWORK
    Xie, Wen
    Wu, Qinzhe
    Ren, Wen
    Zhang, Yuzhuo
    IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, : 7649 - 7652
  • [46] A Multi-scale Convolutional Attention Based GRU Network for Text Classification
    Tang, Xianlun
    Chen, Yingjie
    Dai, Yuyan
    Xu, Jin
    Peng, Deguang
    2019 CHINESE AUTOMATION CONGRESS (CAC2019), 2019, : 3009 - 3013
  • [47] Heterogeneous attention multi-scale network for efficient weld seam classification
    Enpei Guo
    Feixu Chen
    Scientific Reports, 15 (1)
  • [48] Semantic Adversarial Network with Multi-Scale Pyramid Attention for Video Classification
    Xie, De
    Deng, Cheng
    Wang, Hao
    Li, Chao
    Tao, Dapeng
    THIRTY-THIRD AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FIRST INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / NINTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2019, : 9030 - 9037
  • [49] Multi-scale Attention Convolutional Neural Network for time series classification
    Chen, Wei
    Shi, Ke
    NEURAL NETWORKS, 2021, 136 (136) : 126 - 140
  • [50] Multi-scale Fusion Attention Network for Industrial Surface Defect Classification
    Wu, Cong
    Lei, Sicheng
    Xu, Huawei
    Xing, Tongzhen
    PROCEEDINGS OF THE 2024 27 TH INTERNATIONAL CONFERENCE ON COMPUTER SUPPORTED COOPERATIVE WORK IN DESIGN, CSCWD 2024, 2024, : 594 - 599