论文《facial landmark detection by deep multi-task learning》读后小记

本文详细介绍了一种新的深度模型TCDCN,该模型通过多种辅助任务提高特征点定位精度,尤其适用于处理遮挡和大姿态变化的面部图像。TCDCN采用卷积神经网络结构,并针对不同任务设置独立的损失函数,通过早期终止训练方式防止过拟合。

1. 本文提出一种任务驱动型的深度模型(TCDCN —— Tasks-constrained deep convolutional network

该模型思想:通过一些多样的精细的任务,来优化提升特征点定位精度。

该模型优势:1)优于现有的方法,特别是在处理一些遮挡和大姿态的面部图像时。2)与现有的state-of-the-art的级联深度模型方法相比,其动态的降低了模型的复杂度。

(在这里,姿态判断加入的原因:认为特征点定位不是一个孤立的问题,其估计可以被许多多样且微妙的相关因素所影响,例如,当一个小孩笑的时候,他的嘴是很广泛的打开的,有效地发现以及开发这种本质上相关的面部属性,对我们更精确的检测嘴角会很有帮助,再如,有大的姿态旋转的人脸图像,其瞳孔距离更小,这样的姿态信息可以被当做额外的信息资源,来约束特征点估计的解空间。考虑到那些相对合理且相关的任务,孤立的对待特征点定位问题是事与愿违的。)


2. TCDCN简介

通过多任务学习形式,采用卷积神经网络结构,进行特征点定位。

结构:相较于孙祎论文中的结构,本文仅采用其级联的3级共23个网络中第1级的一个网络来实现所有的任务,这里不做任何人脸部件的剖分,只是将整个人脸图像作为输入。

多任务学习:现有的多任务学习方法,一般都是通用同一个损失函数,通常都假设各个任务的困难程度以及收敛率是相同的,但实际上,其并不是一样的,而且,所有任务的迭代学习在执行过程中不会提前终止。而这里的问题并不是这样的,各任务的困难程度以及收敛率并不是相同的,所以,在这里,每个任务都有各自的损失函数,对于不同的任务,其迭代学习执行次数并不相同,会视情况适时终止。这种早期终止(不让误差继续降低至最小值)学习的方式,可以阻止过拟合。


3. TCDCN

3.1 目标

相较于传统的多任务学习(最大化所有任务的性能),我们的目标是优化主要的任务(特征点定位),这里用r表示。


这里, l表示损失函数,A表示除了主要任务——特征点定位之外的所有辅助任务, lambda表示各个任务重要性比重,W表示各个任务的待学习权重矩阵,x为样本输入各任务的特征(这里,所有任务的输入特征都是一样的),不同上标的y表示样本对应任务的目标值。

由于除了特征点定位属于回归问题外,其他所有任务都属于分类问题,所以,目标可以按如下形式表示:


这里,

3.2 网络结构

正向

其中,作为各任务最后一层的输入特征,整个网络正向传播过程图示为:


反向:可详见论文,但是,论文中有一点漏掉一点东西,但大致应该都可以看明白,这里就不详细介绍了。

对于早期终止训练的做法,论文中设计出一种规则,训练迭代过程中实时计算,就可以了,其设计规则的物理意义也有相应的介绍,这里不做解释了,但其中有一点的是,公式中第一项的物理意义是训练误差的趋势,没有仔细研究,还望有所指教。

3.3 效果

所有任务一起训练效果最好,+单任务,则姿态加入效果相对较好。论文中列出了各任务组合情况的效果、早期终止训练和不这样做的效果等,详细见论文图。我想展示下面这张图:


不难发现,其实SDM方法的价值还是相当高的!方法易懂,速度快,计算简单,相对稳定,精确度较高,综合各种因素,性价比很高,有很大发展空间,看好它。

        对于论文中提到作为初始值的应用,都是很好的想法,也都有尝试,效果是不错的,毕竟特征点定位做的好的关键,很大一部分在于给定一个好的初始值(定位上,也盛行这样一句话:一个好的初值胜过一个好的特征,因为,毕竟好的特征是需要探索发现的)。


欢迎交流!


项目链接:点击打开链接

<think>我们正在将Multi-Task-Facial(多任务面部任务)整合到YOLOv11模型和WIDERFACE数据集的训练中。YOLOv11是一个目标检测模型,而WIDERFACE是一个大规模人脸检测数据集。Multi-Task-Facial通常包括人脸检测、关键点定位、表情识别、属性识别等多个任务。目标:在YOLOv11模型的基础上,扩展为多任务模型,同时完成人脸检测(主任务)以及其他面部相关任务(如关键点检测、表情分类等)。步骤:1.**数据准备**:-WIDERFACE数据集主要用于人脸检测,它包含大量的人脸图像和边界框标注。但Multi-Task-Facial需要更多的标注信息(如关键点、表情、属性等)。因此,我们需要对WIDERFACE数据集进行补充标注,或者将WIDERFACE与其他包含多任务标注的数据集(如CelebA,AFLW等)合并使用。-数据格式统一:将多任务标注整合到一个统一的标注文件中(例如,每个图像对应一个标注文件,包含人脸框、关键点坐标、表情标签等)。2.**修改模型架构(YOLOv11)**:-YOLOv11本身是一个单任务模型(检测)。我们需要在其主干网络(backbone)之后添加多个任务特定的分支(heads)。-具体设计:-共享主干网络:使用YOLOv11的主干网络(如CSPDarknet)提取共享特征。-任务分支:-任务1:人脸检测分支(YOLOv11原有的检测头,输出边界框和置信度)。-任务2:人脸关键点检测分支(输出例如68个关键点的坐标)。-任务3:表情分类分支(输出表情类别概率)。-任务4:属性分类分支(如性别、年龄等,每个属性一个二分类或多分类输出)。-注意:每个分支可以是一个或多个卷积层。3.**损失函数**:-总损失是多任务损失的加权和:$$L_{total}=w_{det}\cdotL_{det}+w_{landmark}\cdotL_{landmark}+w_{expr}\cdotL_{expr}+w_{attr}\cdotL_{attr}$$-其中:-$L_{det}$:人脸检测损失,采用YOLOv11原有的损失(包括边界框坐标损失、置信度损失、分类损失)。-$L_{landmark}$:关键点损失,通常使用平滑L1损失或均方误差(MSE)。-$L_{expr}$:表情分类损失,使用交叉熵损失。-$L_{attr}$:属性分类损失,每个属性使用交叉熵损失,然后取平均。-权重选择:可以手动调整,也可以使用动态权重调整方法(如根据任务不确定性[^1])。4.**训练策略**:-联合训练:同时训练所有任务。-数据加载:在数据加载时,需要同时加载所有任务的标注。-训练技巧:-由于不同任务的学习难度和收敛速度不同,可以考虑任务式提前停止[^5]:当某个任务在验证集上性能饱和时,冻结该任务分支,继续训练其他任务。-梯度裁剪:避免不同任务梯度量级差异过大导致训练不稳定。-学习率调整:使用余弦退火等动态调整学习率。5.**评估**:-分别评估每个任务:-人脸检测:使用mAP(meanAveragePrecision)评估。-关键点检测:使用NME(NormalizedMeanError)评估。-表情识别:使用准确率(Accuracy)评估。-属性分类:使用准确率或F1分数评估。6.**推理优化**:-在推理时,我们可以选择只运行需要的任务分支。例如,如果只需要人脸检测,那么只需要运行检测分支;如果需要全部任务,则运行所有分支。关键挑战:-数据集:WIDERFACE没有提供除检测框以外的标注,因此需要额外标注或融合其他数据集。-模型复杂度:多任务模型会增加计算量和参数量,需要考虑模型轻量化或剪枝。代码结构示例(PyTorch伪代码):```pythonclassMultiTaskYOLO(nn.Module):def__init__(self,backbone,num_classes,num_landmarks,num_exprs,num_attrs):super().__init__()self.backbone=backbone#YOLOv11的主干#检测头(YOLO原有的检测头)self.det_head=YOLOHead(...)#关键点头(输出68个关键点,每个点2个坐标)self.landmark_head=nn.Sequential(nn.Conv2d(backbone_out_channels,256,3,padding=1),nn.ReLU(),nn.Conv2d(256,num_landmarks*2,1))#表情分类头(全局平均池化后接全连接层)self.expr_head=nn.Sequential(nn.AdaptiveAvgPool2d(1),nn.Flatten(),nn.Linear(backbone_out_channels,num_exprs))#属性分类头(可以多个属性共享一个头,也可以每个属性单独一个头)self.attr_head=nn.ModuleList([nn.Sequential(nn.AdaptiveAvgPool2d(1),nn.Flatten(),nn.Linear(backbone_out_channels,num_classes_attr[i]))foriinrange(num_attrs)])defforward(self,x):features=self.backbone(x)det_out=self.det_head(features)landmark_out=self.landmark_head(features)expr_out=self.expr_head(features)attr_outs=[head(features)forheadinself.attr_head]returndet_out,landmark_out,expr_out,attr_outs```训练循环中的损失计算示例:```python#假设每个任务的损失函数已经定义loss_det=det_criterion(det_pred,det_target)loss_landmark=landmark_criterion(landmark_pred,landmark_target)loss_expr=expr_criterion(expr_pred,expr_target)loss_attr=0foriinrange(num_attrs):loss_attr+=attr_criterion(attr_preds[i],attr_targets[:,i])loss_attr/=num_attrs#总损失,权重可以根据需要调整total_loss=w_det*loss_det+w_landmark*loss_landmark+w_expr*loss_expr+w_attr*loss_attr```注意事项:-在训练前,需要确保每个任务都有对应的标注数据。对于部分缺失标注的数据,可以采用掩码机制跳过该任务的损失计算。-由于WIDERFACE数据集规模很大,补充标注的工作量巨大,可以考虑使用迁移学习:先在完整标注的小数据集上预训练多任务模型,再用WIDERFACE的检测标注进行微调(此时其他任务分支的标注缺失,可以冻结这些分支,只训练检测分支)。引用说明:-动态权重调整(如根据任务不确定性)可参考[^1]。-任务式提前停止策略可参考[^5]。相关问题:1.如何为WIDERFACE数据集添加人脸关键点、表情等标注?2.多任务训练中,如何平衡不同任务之间的损失权重?3.在YOLOv11多任务模型中,如何优化推理速度?4.如何处理训练数据中部分任务标注缺失的情况?</think>### 在YOLOv11模型中结合WIDER FACE数据集实现Multi-Task-Facial训练的步骤 #### **1. 数据准备与标注转换** - **WIDER FACE数据集处理**: - 下载并解压WIDER FACE数据集(包含32,203张图像和393,703个人脸标注) - 转换标注格式为YOLO格式(归一化坐标): $$ x_{center} = \frac{x + w/2}{img_w}, \quad y_{center} = \frac{y + h/2}{img_h} $$ $$ w_{norm} = \frac{w}{img_w}, \quad h_{norm} = \frac{h}{img_h} $$ - 生成`train.txt`和`val.txt`文件列表 - **多任务标注扩展**: - 为WIDER FACE补充关键点/表情/属性标注(使用预训练模型自动标注或人工标注) - 创建多任务标注文件(每行格式): ``` image_path x1,y1,x2,y2,conf,landmark_x1,landmark_y1,...,expr_id,attr1,attr2,... ``` #### **2. 模型架构修改** - **主干网络共享**: ```python # YOLOv11主干网络(CSPDarknet) backbone = CSPDarknet(depth_multiple=1.0, width_multiple=1.0) ``` - **多任务分支扩展**: ```python # 检测分支(原始YOLO头) detect_head = Detect(nc=1) # 仅人脸检测,nc=1 # 关键点分支(68个点) landmark_head = nn.Sequential( Conv(1024, 256, 3), nn.Flatten(), nn.Linear(256*7*7, 136) # 68*2=136 ) # 表情分支(7类基本表情) expr_head = nn.Sequential( nn.AdaptiveAvgPool2d(1), nn.Flatten(), nn.Linear(1024, 7) ) ``` #### **3. 损失函数设计** - **多任务加权损失**: $$ \mathcal{L}_{total} = \lambda_{det}\mathcal{L}_{det} + \lambda_{landmark}\mathcal{L}_{landmark} + \lambda_{expr}\mathcal{L}_{expr} $$ 其中: - $\mathcal{L}_{det}$:YOLO检测损失(CIoU + 置信度)[^1] - $\mathcal{L}_{landmark}$:平滑L1关键点损失 - $\mathcal{L}_{expr}$:表情分类交叉熵损失 - **动态权重调整**(推荐): $$ \lambda_i = \frac{1}{2\sigma_i^2}, \quad \mathcal{L}_{total} = \sum_i \frac{\mathcal{L}_i}{\sigma_i^2} + \log \sigma_i $$ 通过可学习的$\sigma_i$自动平衡任务权重[^2] #### **4. 训练流程实现** 1. **数据加载**: ```python dataset = MultiTaskDataset( img_dir="WIDER_train/images", annot_file="multitask_annotations.txt", transforms=augmentations # 包含Mosaic, MixUp等 ) ``` 2. **训练循环**: ```python for imgs, targets in dataloader: # targets: [batch_idx, cls, box, landmarks, expr] det_out, landmark_out, expr_out = model(imgs) # 计算多任务损失 loss_det = compute_yolo_loss(det_out, targets[..., :6]) loss_landmark = smooth_l1(landmark_out, targets[..., 6:142]) loss_expr = F.cross_entropy(expr_out, targets[..., 142]) # 动态加权 total_loss = loss_det/(2*sigma_det**2) + loss_landmark/(2*sigma_landmark**2) + loss_expr/(2*sigma_expr**2) total_loss += torch.log(sigma_det*sigma_landmark*sigma_expr) total_loss.backward() optimizer.step() ``` #### **5. 关键训练技巧** - **渐进式训练**: 1. 第一阶段:仅训练检测分支(冻结其他分支) 2. 第二阶段:联合训练所有分支 3. 第三阶段:微调任务特定分支 - **梯度处理**: ```python # 梯度裁剪防止冲突 torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=10.0) # 梯度归一化(可选) for param in model.parameters(): if param.grad is not None: param.grad.data = param.grad.data / (param.grad.data.norm() + 1e-8) ``` - **数据增强策略**: - 基础增强:随机裁剪、色彩抖动、旋转(±30°) - 高级增强:CutMix, GridMask - **关键点增强**:弹性形变(确保关键点同步变换) #### **6. 评估与验证** - **分任务评估指标**: | 任务 | 指标 | 说明 | |--------------|-----------------------|--------------------------| | 人脸检测 | mAP@0.5 | WIDER FACE官方评估标准 | | 关键点检测 | NME (Normalized Mean Error) | $$ \frac{1}{N}\sum_i \frac{\|pred_i - gt_i\|_2}{d_{interocular}} $$ | | 表情识别 | 准确率(Accuracy) | 7类表情分类准确率 | --- ### 部署优化建议 - **推理加速**: ```python # 仅启用必要分支 model.set_task('detection') # 仅运行检测头 detections = model(imgs) # 对检测结果运行其他任务 for bbox in detections: landmark = model.run_head('landmark', roi_crop(img, bbox)) ``` - **模型轻量化**: - 使用通道剪枝压缩共享主干网络 - 量化INT8推理(TensorRT部署) --- ### 相关问题 1. 如何在WIDER FACE数据集上自动生成多任务标注? 2. YOLOv11多任务训练中如何解决不同任务收敛速度不一致的问题? 3. 动态损失权重$\sigma$的初始化值应该如何设置? 4. 多任务模型中如何平衡检测精度与关键点精度? [^1]: 检测损失包含CIoU损失、置信度损失和分类损失,是人脸检测的基础 [^2]: 动态权重调整通过可学习的$\sigma$参数自动平衡多任务损失权重
评论 7
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值