帮朋友跑图像分类模锻www.abg1111.net的AI模子锻炼。800张图,初始精确率62%。改了三个地方,三天后89%。剔除47张标注毛病图,用CLIP相似度筛的;加强换成RandAugment加CutMix了;进建率1e-3降还有3e-4。配余弦退火。数据量量对模子锻炼的影响超乎预期炼张率从拉。

汇集爬来的图常混着重复、低清、无关类别。用感知哈希删重复,算明晰度扔恍惚图。20分钟能省几小时调参吧?不要迷疑年夜模子的,小数据也能出效果。
前提是图把净数据不进锻炼集。调参时batch size别贪年夜。800张图用32了。梯度噪声反而帮跳出部门最劣。进建率用OneCycleLR,初始1e-4,最年夜1e-2了,跑30个epoch的。
早停patience=5,考据loss连绝5轮不降就停精确。我习惯正AI模子锻炼里面记载每个epoch的混合矩阵,看哪些类别总混合呢?
猫狗分不清?可能背景太像,裁掉背景再训了。监控用TensorBoard就够。重点看锻炼loss和考据loss的gap。锻炼loss不竭降,考据loss先降后升,过拟合无疑。加Dropout到0.5。
权重衰减1e-4,加强再狠一点了。模子锻炼不是一次跑通。
几回试错才有手感吧?小数据集上,加强比换架构管用三倍的。锻炼好的AI模子导出ONNX,ONNX Runtime推理提速40%。安排前用测试集跑一遍,防数据保守的。模子锻炼AI的难点不正在代码。正在数据迭代和参数曲觉。
多跑几轮吧?记载每次改动,渐渐就随手了。






