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 条
  • [1] XMP: A CROSS-ATTENTION MULTI-SCALE PERFORMER FOR FILE FRAGMENT CLASSIFICATION
    Park, Jeong Gyu
    Liu, Sisung
    Hong, Je Hyeong
    2024 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING, ICASSP 2024, 2024, : 4505 - 4509
  • [2] CrossFormer: Multi-scale cross-attention for polyp segmentation
    Chen, Lifang
    Ge, Hongze
    Li, Jiawei
    IET IMAGE PROCESSING, 2023, 17 (12) : 3441 - 3452
  • [3] MSCAN: multi-scale self- and cross-attention network for RNA methylation site prediction
    Honglei Wang
    Tao Huang
    Dong Wang
    Wenliang Zeng
    Yanjing Sun
    Lin Zhang
    BMC Bioinformatics, 25
  • [4] A VAN-Based Multi-Scale Cross-Attention Mechanism for Skin Lesion Segmentation Network
    Liu, Shuang
    Zhuang, Zeng
    Zheng, Yanfeng
    Kolmanic, Simon
    IEEE ACCESS, 2023, 11 : 81953 - 81964
  • [5] Multi-scale network with shared cross-attention for audio–visual correlation learning
    Jiwei Zhang
    Yi Yu
    Suhua Tang
    Wei Li
    Jianming Wu
    Neural Computing and Applications, 2023, 35 : 20173 - 20187
  • [6] MSCAN: multi-scale self- and cross-attention network for RNA methylation site prediction
    Wang, Honglei
    Huang, Tao
    Wang, Dong
    Zeng, Wenliang
    Sun, Yanjing
    Zhang, Lin
    BMC BIOINFORMATICS, 2024, 25 (01)
  • [7] Cross-Attention Enhanced Pyramid Multi-Scale Networks for Sensor-Based Human Activity Recognition
    Pang, Hongsen
    Zheng, Li
    Fang, Hongbin
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2024, 28 (05) : 2733 - 2744
  • [8] Multi-scale network with shared cross-attention for audio-visual correlation learning
    Zhang, Jiwei
    Yu, Yi
    Tang, Suhua
    Li, Wei
    Wu, Jianming
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (27) : 20173 - 20187
  • [9] Multi-Scale Spatial-Temporal Attention Networks for Functional Connectome Classification
    Kong, Youyong
    Zhang, Xiaotong
    Wang, Wenhan
    Zhou, Yue
    Li, Yueying
    Yuan, Yonggui
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2025, 44 (01) : 475 - 488
  • [10] Enhancing Medical Image Classification With Context Modulated Attention and Multi-Scale Feature Fusion
    Zhang, Renhan
    Luo, Xuegang
    Lv, Junrui
    Cao, Junyang
    Zhu, Yangping
    Wang, Juan
    Zheng, Bochuan
    IEEE ACCESS, 2025, 13 : 15226 - 15243