【特征】图的故事
22-Android Malware-Contrastive Learning
Idea
在安卓恶意软件检测中使用对比学习对抗混淆问题。
- 先获取函数调用图，再用中心性分析(centrality analysis)，将调用图转换为图像。
- 用对比学习最大化正样本的相似性，最小化负样本的相似性，减小代码混淆引入的差异，扩大不同类别恶意程序的差异，让混淆后的程序也能正确分类。
- 混淆后的样本视为原始样本的正样本之一。
对比学习后的可视化和解释性问题：
- 获取热力图：“Gradient-weighted Class Activation Mapping++ (Grad-CAM++)”
“GradCAM++ is a class-discriminative localization technique that generates visual explanations for any CNN-based network without changing the architecture or retraining.”(Wu 等, 2022, p. 2)
框架
- 对Android Malware静态分析得到函数调用图，进入Image Generation阶段。
Image Generation：
- 下图为426个敏感API（先验），是由三个API集合并得到。
- 社交网络的分析方法：对于每个API调用，用四种中心性计算方法，计算节点在网络（函数调用图）中的重要性。即对一个API调用（一行），计算得到四列（四种中心性），最后得到426*4的向量。若API未在程序中出现过，填0.
- 对426*4的向量在末尾添加60个0，即填充15行，reshape为42*42的尺寸。
- 为什么是4242：因为是4的整数倍，用4264填充减少。
Contrastive Learning for Classification:
有监督对比学习用于分类：
有监督对比学习，比对常见分类模型有两个主要改动：
- 数据增强
- 损失函数
此处正样本对不是数据增强的结果，属于同一类的样本为正样本。在一个batch内有许多正样本和许多负样本。
其实就是改个损失函数...
Discussion
- 用社交网络的中心性分析方法不错。
- 对比学习约等于改个损失函数，涨点的方法，并没有那么高大上。
- 可扩展性一般，426敏感API 这个数字属于先验，如果新增加API需要进行较大改动。
- 差评：不开源
Reference
Wu, Yueming, et al. "Contrastive Learning for Robust Android Malware Familial Classification." IEEE Transactions on Dependable and Secure Computing (2022).
22-Code Vulnerability-VulCNN
Idea
Input:
- source code
Graph Extraction:
- 提取程序依赖图(PDG)
Sentence Embedding：
- 用Sentence2vec方法，对每个节点都embedding得到节点向量。
Image Generation:
- 计算节点的度和度中心性
- 每种度中心性得到一张特征图，3种度中心性得到一个RGB图像
- 再用CNN
其中需要考虑：
- 节点的个数问题，即每个函数的代码行数，超过一个阈值要裁剪，小于一个阈值要填充。
Discussion
- CNN需要输入固定尺度的图像，故而仍需要限制网络中的节点个数。
- 可以取和sentence embedding维度相同的节点个数，得到一个矩形图像，结合SPPnet，可进行多尺度处理。
- 好评：开源，https://github.com/CGCL-codes/VulCNN
Reference
Y. Wu, D. Zou, S. Dou, W. Yang, D. Xu and H. Jin, "VulCNN: An Image-inspired Scalable Vulnerability Detection System," 2022 IEEE/ACM 44th International Conference on Software Engineering (ICSE), Pittsburgh, PA, USA, 2022, pp. 2365-2376, doi: 10.1145/3510003.3510229.
标签：故事,特征,Image,样本,API,2022,节点,426 From： https://www.cnblogs.com/handsome6/p/features-the-story-of-the-picture-qqmnv.html