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 条
  • [31] Adaptive multi-scale attention convolution neural network for cross-domain fault diagnosis
    Shao, Xiaorui
    Kim, Chang-Soo
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 236
  • [32] Multi-scale Cross Dual Attention Network for Building Change Detection in Remote Sensing Images
    Zhang J.
    Yan Z.
    Ma S.
    Journal of Geo-Information Science, 2023, 25 (12) : 2487 - 2500
  • [33] Cross-Attention Based Multi-Resolution Feature Fusion Model for Self-Supervised Cervical OCT Image Classification
    Wang, Qingbin
    Chen, Kaiyi
    Dou, Wanrong
    Ma, Yutao
    IEEE-ACM TRANSACTIONS ON COMPUTATIONAL BIOLOGY AND BIOINFORMATICS, 2023, 20 (04) : 2541 - 2554
  • [34] Residual Self-Calibrated Network With Multi-Scale Channel Attention for Accurate EOG-Based Eye Movement Classification
    Zeng, Zheng
    Tao, Linkai
    Su, Ruizhi
    Tuheti, Adili
    Huang, Hao
    Chen, Chen
    Chen, Wei
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2025, 29 (01) : 118 - 127
  • [35] Enhancing few-shot image classification through learnable multi-scale embedding and attention mechanisms
    Askari, Fatemeh
    Fateh, Amirreza
    Mohammadi, Mohammad Reza
    NEURAL NETWORKS, 2025, 187
  • [36] Multi-Scale Residual Spectral-Spatial Attention Combined with Improved Transformer for Hyperspectral Image Classification
    Wang, Aili
    Zhang, Kang
    Wu, Haibin
    Iwahori, Yuji
    Chen, Haisong
    ELECTRONICS, 2024, 13 (06)
  • [37] Multi-Scale Convolutional Attention and Riemannian Geometry Network for EEG-Based Motor Imagery Classification
    Zhou, Ben
    Wang, Lei
    Xu, Wenchang
    Jiang, Chenyu
    IEEE ACCESS, 2024, 12 : 79731 - 79740
  • [38] Multi-Scale Cross-Band Encoding of Sectored Local Binary Pattern for Robust Texture Classification
    Song, Tiecheng
    Luo, Lin
    Xin, Liangliang
    Gao, Chenqiang
    2018 24TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2018, : 1163 - 1168
  • [39] Cross attention guided multi-scale feature fusion for false-positive reduction in pulmonary nodule detection
    Gu, Zhongxuan
    Li, Yueyang
    Luo, Haichi
    Zhang, Caidi
    Du, Hongqun
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 151
  • [40] Multi-channel and multi-scale separable dilated convolutional neural network with attention mechanism for flue-cured tobacco classification
    Xu, Ming
    Gao, Jinfeng
    Zhang, Zhong
    Guo, Xin
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (21) : 15511 - 15529