将文件片段转为图像,用视觉模型提升多媒体类型识别准确率
ByteNet: Rethinking Multimedia File Fragment Classification through Visual Perspectives
- 把字节内信息转成2D灰度图,用视觉网络同时捕捉字节间和字节内关联
- 在14个测试案例中最高提升12.2%准确率,优于现有方法
- 适合需要高精度文件类型识别的存储与通信系统
多媒体文件片段分类(MFFC)旨在无系统元数据情况下识别文件片段类型(如图像、视频、音频、文本),对多媒体存储与通信至关重要。现有方法通常将片段视为一维字节序列,关注字节间的相互关系(interbytes),却忽视了字节内部的信息(intrabytes)。本文首次提出Byte2Image模型,将被忽略的字节内信息转化为2D灰度图像,重新诠释文件片段。该模型采用滑动字节窗口揭示字节内细节,并通过行堆叠字节ngram将片段嵌入二维空间。由此,复杂字节间与字节内关联可由强大视觉网络共同挖掘。此外,提出端到端双分支网络ByteNet,融合原始1D字节序列与转换后的2D图像:浅层字节分支(BBFE)使用单层全连接层自适应识别特定字节共现模式;深层图像分支(IBFE)基于视觉Transformer有效挖掘复杂关联。在两个代表性基准上14个案例的实验表明,本方法在不同场景下性能优于当前最优方法,最高提升达12.2%。
原文摘要 · Abstract (English)
Multimedia file fragment classification (MFFC) aims to identify file fragment types, e.g., image/video, audio, and text without system metadata. It is of vital importance in multimedia storage and communication. Existing MFFC methods typically treat fragments as 1D byte sequences and emphasize the relations between separate bytes (interbytes) for classification. However, the more informative relations inside bytes (intrabytes) are overlooked and seldom investigated. By looking inside bytes, the bit-level details of file fragments can be accessed, enabling a more accurate classification. Motivated by this, we first propose Byte2Image, a novel visual representation model that incorporates previously overlooked intrabyte information into file fragments and reinterprets these fragments as 2D grayscale images. This model involves a sliding byte window to reveal the intrabyte information and a rowwise stacking of intrabyte ngrams for embedding fragments into a 2D space. Thus, complex interbyte and intrabyte correlations can be mined simultaneously using powerful vision networks. Additionally, we propose an end-to-end dual-branch network ByteNet to enhance robust correlation mining and feature representation. ByteNet makes full use of the raw 1D byte sequence and the converted 2D image through a shallow byte branch feature extraction (BBFE) and a deep image branch feature extraction (IBFE) network. In particular, the BBFE, composed of a single fully-connected layer, adaptively recognizes the co-occurrence of several some specific bytes within the raw byte sequence, while the IBFE, built on a vision Transformer, effectively mines the complex interbyte and intrabyte correlations from the converted image. Experiments on the two representative benchmarks, including 14 cases, validate that our proposed method outperforms state-of-the-art approaches on different cases by up to 12.2%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。