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] Multivariate multi-scale Gaussian for microarray unsupervised classification
    King, Amelia
    Yang, ZiHua
    Yang, Zheng Rong
    PROCEEDINGS OF THE 2014 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2014, : 2458 - 2463
  • [22] Multi-Scale Self-Attention for Text Classification
    Guo, Qipeng
    Qiu, Xipeng
    Liu, Pengfei
    Xue, Xiangyang
    Zhang, Zheng
    THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 7847 - 7854
  • [23] Multi-Scale Attention Network for Diabetic Retinopathy Classification
    Al-Antary, Mohammad T.
    Arafa, Yasmine
    IEEE ACCESS, 2021, 9 : 54190 - 54200
  • [24] Multi-Scale and Attention based ResNet for Heartbeat Classification
    Zhang, Haojie
    Yang, Gongping
    Huang, Yuwen
    Yuan, Feng
    Yin, Yilong
    2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 1529 - 1535
  • [25] MSANet: Multi-scale attention networks for image classification
    Cao, Ping
    Xie, Fangxin
    Zhang, Shichao
    Zhang, Zuping
    Zhang, Jianfeng
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (24) : 34325 - 34344
  • [26] MSANet: Multi-scale attention networks for image classification
    Ping Cao
    Fangxin Xie
    Shichao Zhang
    Zuping Zhang
    Jianfeng Zhang
    Multimedia Tools and Applications, 2022, 81 : 34325 - 34344
  • [27] Multi-label Thoracic Disease Image Classification with Cross-Attention Networks
    Ma, Congbo
    Wang, Hu
    Hoi, Steven C. H.
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT VI, 2019, 11769 : 730 - 738
  • [28] LQCANet: Learnable-Query-Guided Multi-Scale Fusion Network Based on Cross-Attention for Radar Semantic Segmentation
    Zhuang, Long
    Jiang, Tiezhen
    Jiang, Hao
    Wang, Anqi
    Huang, Zhixiang
    IEEE TRANSACTIONS ON INTELLIGENT VEHICLES, 2024, 9 (02): : 3330 - 3344
  • [29] 3D lymphoma segmentation on PET/CT images via multi-scale information fusion with cross-attention
    Huang, Huan
    Qiu, Liheng
    Yang, Shenmiao
    Li, Longxi
    Nan, Jiaofen
    Li, Yanting
    Han, Chuang
    Zhu, Fubao
    Zhao, Chen
    Zhou, Weihua
    MEDICAL PHYSICS, 2025,
  • [30] Multi-task Hierarchical Cross-Attention Network for Multi-label Text Classification
    Lu, Junyu
    Zhang, Hao
    Shen, Zhexu
    Shi, Kaiyuan
    Yang, Liang
    Xu, Bo
    Zhang, Shaowu
    Lin, Hongfei
    NATURAL LANGUAGE PROCESSING AND CHINESE COMPUTING, NLPCC 2022, PT II, 2022, 13552 : 156 - 167