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] Multi-scale context feature and cross-attention network-enabled system and software-based for pavement crack detection
    Wen, Xin
    Li, Shuo
    Yu, Hao
    He, Yu
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 127
  • [32] Multi-scale self-attention mixup for graph classification *
    Kong, Youyong
    Li, Jiaxing
    Zhang, Ke
    Wu, Jiasong
    PATTERN RECOGNITION LETTERS, 2023, 168 : 100 - 106
  • [33] Multi-Scale Dense Graph Attention Network for Hyperspectral Classification
    Wang, Chen
    Li, Lu
    Wang, Zhongqi
    Ma, Jingyao
    Kong, Yunlong
    Wang, Yanfeng
    Chang, Jianrui
    Zhang, Zimeng
    Lin, Xinyu
    CANADIAN JOURNAL OF REMOTE SENSING, 2024, 50 (01)
  • [34] Multi-Scale Attention-Guided Network for mammograms classification
    Xu, Chunbo
    Lou, Meng
    Qi, Yunliang
    Wang, Yiming
    Pi, Jiande
    Ma, Yide
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 68
  • [35] Pulmonary Textures Classification via a Multi-Scale Attention Network
    Xu, Rui
    Cong, Zhen
    Ye, Xinchen
    Hirano, Yasushi
    Kido, Shoji
    Gyobu, Tomoko
    Kawata, Yutaka
    Honda, Osamu
    Tomiyama, Noriyuki
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2020, 24 (07) : 2041 - 2052
  • [36] Multi-scale Contrastive Learning with Attention for Histopathology Image Classification
    Tan, Jing Wei
    Khoa Tuan Nguyen
    Lee, Kyoungbun
    Jeong, Won-Ki
    MEDICAL IMAGING 2023, 2023, 12471
  • [37] An Enhanced MobileNet with Multi-scale Attention Aggregation for DR Classification
    Xi, Heran
    Ji, Hongxu
    Hu, Yang
    Li, Jinbao
    Zhu, Jinghua
    WEB AND BIG DATA, APWEB-WAIM 2024, PT I, 2024, 14961 : 376 - 390
  • [38] Cross Attention-Based Multi-Scale Convolutional Fusion Network for Hyperspectral and LiDAR Joint Classification
    Ge, Haimiao
    Wang, Liguo
    Pan, Haizhu
    Liu, Yanzhong
    Li, Cheng
    Lv, Dan
    Ma, Huiyu
    REMOTE SENSING, 2024, 16 (21)
  • [39] Pulp grade monitoring using binocular image through multi-scale feature cross-attention fusion network and saliency map constraint
    Zhong, Yuze
    Tang, Zhaohui
    Zhang, Hu
    Dai, Zhien
    Nie, Zibang
    Xie, Yongfang
    ADVANCED ENGINEERING INFORMATICS, 2024, 62
  • [40] Blind Gaussian Deep Denoiser Network using Multi-Scale Pixel Attention
    Thakur, Ramesh Kumar
    Maji, Suman Kumar
    2022 IEEE INTERNATIONAL CONFERENCE ON VISUAL COMMUNICATIONS AND IMAGE PROCESSING (VCIP), 2022,