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 条
  • [21] Cross-attention multi-branch network for fundus diseases classification using SLO images
    Xie, Hai
    Zeng, Xianlu
    Lei, Haijun
    Du, Jie
    Wang, Jiantao
    Zhang, Guoming
    Cao, Jiuwen
    Wang, Tianfu
    Lei, Baiying
    MEDICAL IMAGE ANALYSIS, 2021, 71
  • [22] A Multi-Scale Progressive Collaborative Attention Network for Remote Sensing Fusion Classification
    Ma, Wenping
    Li, Yating
    Zhu, Hao
    Ma, Haoxiang
    Jiao, Licheng
    Shen, Jianchao
    Hou, Biao
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (08) : 3897 - 3911
  • [23] Hyperspectral Image Classification Based on Multi-Scale Convolutional Features and Multi-Attention Mechanisms
    Sun, Qian
    Zhao, Guangrui
    Xia, Xinyuan
    Xie, Yu
    Fang, Chenrong
    Sun, Le
    Wu, Zebin
    Pan, Chengsheng
    REMOTE SENSING, 2024, 16 (12)
  • [24] Cross-supervised Crowd Counting via Multi-scale Channel Attention
    Yang, Kexin
    Luan, Fangjun
    Yuan, Shuai
    Liu, Guoqi
    INFORMATION TECHNOLOGY AND CONTROL, 2024, 53 (03): : 785 - 797
  • [25] A Transformer utilizing bidirectional cross-attention for multi-modal classification of Age-Related Macular Degeneration
    Li, Jianfeng
    Wang, Zongda
    Chen, Yuanqiong
    Zhu, Chengzhang
    Xiong, Mingqiang
    Bai, Harrison Xiao
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 109
  • [26] Multi-Scale and Multi-Factor ViT Attention Model for Classification and Detection of Pest and Disease in Agriculture
    Xie, Mingyao
    Ye, Ning
    APPLIED SCIENCES-BASEL, 2024, 14 (13):
  • [27] An Attention-Based 3D CNN With Multi-Scale Integration Block for Alzheimer's Disease Classification
    Wu, Yuanchen
    Zhou, Yuan
    Zeng, Weiming
    Qian, Qian
    Song, Miao
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2022, 26 (11) : 5665 - 5673
  • [28] A diagonal masking self-attention-based multi-scale network for motor imagery classification
    Yang, Kaijun
    Wang, Jihong
    Yang, Liantao
    Bian, Lifeng
    Luo, Zijiang
    Yang, Chen
    JOURNAL OF NEURAL ENGINEERING, 2024, 21 (03)
  • [29] Weld TOFD defect classification method based on multi-scale CNN and cascaded focused attention
    Tang, Donglin
    Zhang, Junhui
    Wang, Pingjie
    He, Yuanyuan
    JOURNAL OF MANUFACTURING PROCESSES, 2025, 138 : 157 - 168
  • [30] Few-shot Image Classification Algorithm Based on Multi-scale Attention and Residual Network
    Wang, Qi
    Jin, Huazhong
    Yan, Meng
    Li, Lin
    2023 3RD ASIA-PACIFIC CONFERENCE ON COMMUNICATIONS TECHNOLOGY AND COMPUTER SCIENCE, ACCTCS, 2023, : 641 - 645