UniFramework 01
文章的部分内容被密码保护: --- DON'T MODIFY THIS LINE --- @aiken 2021 Framework Abstract Try To make structure universal,编写一个自己的通用的架构,框架化,满足通过不同的model文件和特殊配置文件就能实现不同的模型的一个架构。 只是一个初步的框架集成,还有很多没有完善的地方,目前测试了ResNet18 跑Cifar10,没有什么问题,如果有什么可以改进的地方,或者你实现了一些Feature,*欢迎进行交流*!(私下联系我最好啦!) 感谢帮助 还有一些可以参数化或者可视化的地方,由于时间关系目前还没有修改,有兴趣的可以自己先添加一下 暂时只集成了分类的模块,后续可能会随缘扩展 本框架主要希望实现的是:易读性,可拓展性,以及简洁; 希望将重要的,可变的参数都尽量的分离出来,通过配置文件和命令行参数去定义和运行我们的网络,在这种情况下实现一个较好的工作流程。 Final Project Design PURPOSE:新类发现和模型自主更新;同时希望能够解决长尾分布的数据情景; **ANALYSIS:**为了实现这种模型的自主更新过程,将整体的流程分成两个部分 启动(start): self supervissed 等方法无监督的学习特征提取网络(这种方式是否会对Unbalance产生增益) 初始化预测模型: 基于Unbalance的数据训练一个基础的分类模型,在输出分类结果的同时需要输出对应的预测置信度,这两个其实都是一些简单的Trick,而最重要的是Backbone的分类效果需要得到保证,同时Backbone需要支撑后续的模型蒸馏更新。 模型的自主更新和迭代: Online:在线运行推断模型,通过置信度输出筛选出新类样本,将样本在样本池中收集 Offline:基于样本池的规模和评估触发离线更新:伪标签生成模型;模型蒸馏和更新 创新点:自主新类发现和学习 ...