Scattering-ViT
基于可学习散射网络和 Vision Transformer 的射频信号分类框架
2026年4月
GitHub
PythonPyTorchViTScattering Transform
项目简介#
实现了一个前沿深度学习框架,结合可学习散射网络和 Vision Transformer 进行射频信号分类。处理频谱图数据(射频信号的时频表示),实现多信号类型的高精度分类。
核心功能#
- 多架构支持:ResNet50、ViT、DeiT、PVT、Swin Transformer 及混合模型
- 可学习散射网络:可训练的小波散射变换,鲁棒特征提取
- 多尺度处理:S0、S1、S2 多阶散射系数,多分辨率分析
- 生产就绪:混合精度训练、梯度累积、学习率调度
技术栈#
- Python / PyTorch 2.x / Kymatio
- timm (Vision Transformers) / TensorBoard
- scikit-learn / matplotlib / seaborn