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 条
  • [41] Multi-channel and multi-scale separable dilated convolutional neural network with attention mechanism for flue-cured tobacco classification
    Ming Xu
    Jinfeng Gao
    Zhong Zhang
    Xin Guo
    Neural Computing and Applications, 2023, 35 : 15511 - 15529
  • [42] Triplet-Metric-Guided Multi-Scale Attention for Remote Sensing Image Scene Classification with a Convolutional Neural Network
    Wang, Hong
    Gao, Kun
    Min, Lei
    Mao, Yuxuan
    Zhang, Xiaodian
    Wang, Junwei
    Hu, Zibo
    Liu, Yutong
    REMOTE SENSING, 2022, 14 (12)
  • [43] DAMS-Net: Dual attention and multi-scale information fusion network for 12-lead ECG classification
    Zhou, Rongzhou
    Yao, Junfeng
    Hong, Qingqi
    Zheng, Yuan
    Zheng, Liling
    METHODS, 2023, 220 : 134 - 141
  • [44] MSRA-G: Combination of multi-scale residual attention network and generative adversarial networks for hyperspectral image classification
    Zhao, Jinling
    Hu, Lei
    Huang, Linsheng
    Wang, Chuanjian
    Liang, Dong
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 121
  • [45] A Multi-Scale Cross-Fusion Medical Image Segmentation Network Based on Dual-Attention Mechanism Transformer
    Cui, Jianguo
    Wang, Liejun
    Jiang, Shaochen
    APPLIED SCIENCES-BASEL, 2023, 13 (19):
  • [46] Research on a phonocardiogram and electrocardiogram signal classification algorithm based on global group coordinate attention mechanism and multi-scale feature fusion
    Wang, Guofu
    Yang, Yuhua
    Cui, Jiangong
    Zhang, Wendong
    Zhang, Guojun
    Wang, Renxin
    Shi, Pengcheng
    Tian, Hua
    SENSOR REVIEW, 2025, : 399 - 412
  • [47] Incorporating Attention Mechanism, Dense Connection Blocks, and Multi-Scale Reconstruction Networks for Open-Set Hyperspectral Image Classification
    Zhou, Huaming
    Wu, Haibin
    Wang, Aili
    Iwahori, Yuji
    Yu, Xiaoyu
    REMOTE SENSING, 2023, 15 (18)
  • [48] Multi-scale attention-based adaptive feature fusion network for fine-grained ship classification in remote sensing scenarios
    Liu, Kun
    Zhang, Xiaomeng
    Xu, Zhijing
    Liu, Sidong
    JOURNAL OF APPLIED REMOTE SENSING, 2024, 18 (03)
  • [49] Dual-Path and Multi-Scale Enhanced Attention Network for Retinal Diseases Classification Using Ultra-Wide-Field Images
    Chen, Fangsheng
    Ma, Shaodong
    Hao, Jinkui
    Liu, Mengting
    Gu, Yuanyuan
    Yi, Quanyong
    Zhang, Jiong
    Zhao, Yitian
    IEEE ACCESS, 2023, 11 : 45405 - 45415
  • [50] M2SSCENet: a multi-branch multi-scale network with spatial-spectral cross-enhancement for hyperspectral and LiDAR data classification
    Yu, Changhong
    Zhang, Mingxuan
    JOURNAL OF SUPERCOMPUTING, 2025, 81 (04)