Enhancing Accuracy of Uncertainty Estimation in Appearance-based Gaze Tracking with Probabilistic Evaluation and Calibration¶
会议: CVPR 2026 arXiv: 2501.14894 代码: 有(项目主页提供) 领域: 视频理解 关键词: 视线追踪, 不确定性估计, 后验校准, 域偏移, 覆盖概率误差
一句话总结¶
提出一种数据高效的后验校准方法,通过等保序回归将不确定性感知视线追踪模型的预测分布与真实观测分布对齐,并引入 Coverage Probability Error (CPE) 指标替代不可靠的误差-不确定性相关性(EUC)来评估不确定性质量。
研究背景与动机¶
基于外观的视线追踪(appearance-based gaze tracking)通过深度学习直接从眼部/面部图像预测注视角度,是驾驶监控等安全关键应用的核心技术。仅给出点估计不够,系统需要知道"这次估计有多可靠",因此不确定性估计至关重要。
现有痛点: 1. 域偏移导致不确定性失准:现有不确定性感知模型(heteroskedastic regression、quantile regression 等)的不确定性估计仅在训练域内有效;当测试数据的光照、相机、受试者等发生变化时,预测的方差数值变得不可靠——模型可能对错误预测给出过高置信度 2. 评价指标有缺陷:广泛使用的 EUC(Error-Uncertainty Correlation)假设不确定性与预测误差相关,但这是虚假相关——不确定性来源于 aleatoric 和 epistemic 因素而非预测误差本身,因此 EUC 无法可靠评估不确定性质量 3. 参数级适应代价高:meta-learning 或 transfer learning 可以纠正不确定性,但需要大量目标域数据重新学习条件分布
核心 idea:不修改模型参数,用后验输出校准(post-hoc calibration)——学一个单调映射函数将名义概率映射到实际覆盖概率,使校准后的分布逼近真实分布。仅需约 50 个校准样本。
方法详解¶
整体框架¶
给定一个已训练的不确定性感知视线追踪模型 \(H\),它对输入图像 \(x_t\) 输出高斯分布(均值=注视角度,方差=不确定性)。校准流程在模型输出后附加一个校准回归器 \(R\),将不准确的 CDF 映射到校准后 CDF \(R \circ F\)。
关键设计¶
- Coverage Probability Error (CPE) 指标:
- 做什么:评估不确定性模型的校准质量
- 核心思路:对每个名义概率 \(p\),计算真实标签落在对应分位数以下的实际比例 \(\hat{P}(p)\),理想情况应有 \(\hat{P}(p) = p\)。CPE 定义为全概率范围上偏差的 RMSE: $\(CPE = \sqrt{\frac{1}{n}\sum_{i=0}^{n} p_{err}\left(\frac{i}{n}\right)^2}, \quad p_{err}(p) = \left|p - \hat{P}(p)\right|\)$
-
设计动机:EUC 依赖误差与不确定性的虚假相关,CPE 直接度量预测分布与观测分布的匹配度,是 proper scoring rule 的思路。例如 CPE=0.05 意味着名义 80% 置信区间实际覆盖约 70-90%
-
等保序回归校准器(Isotonic Regression Calibrator):
- 做什么:学习一个单调递增映射 \(R: [0,1] \to [0,1]\),将名义概率映射到误差补偿后的概率
- 核心思路:收集校准数据集 \(D = \{(p_i, \hat{P}(p_i))\}\),用 isotonic regression 拟合 \(R\),满足单调性约束 \(R(p_i) \leq R(p_{i+1})\)。测试时用 \(\tilde{p} = R(p)\) 替代名义概率 \(p\) 来计算分位数: $\(\frac{\sum_{t=1}^{T} I\{\theta_t \leq F_t^{-1}(R(p))\}}{T} \to p\)$
-
设计动机:(1) 非参数化方法,不对校准函数施加特定参数形式假设(如 temperature scaling 假设线性关系),能捕捉非线性失准;(2) 等保序回归天然满足 CDF 的单调性要求;(3) 仅需约 50 个校准样本,数据效率高;(4) 不修改原模型参数,适用于任何输出概率分布的模型
-
分轴独立校准:
- 做什么:对注视角度的 yaw 和 pitch 分量分别进行校准
- 核心思路:视线的水平(yaw)和垂直(pitch)分量具有不同的误差分布特性,独立校准可以更精确地匹配各轴的真实分布
- 设计动机:两个轴的域偏移严重程度可能不同(如水平方向受头部偏转影响更大),分开校准更灵活
损失函数 / 训练策略¶
- 基础模型使用 heteroskedastic NLL loss 训练:\(NLL_t = \frac{1}{2}\ln(\hat{\sigma}_t^2) + \frac{l_{n,t}}{2\hat{\sigma}_t^2}\),其中 \(l_{n,t}\) 为 smooth L1 loss
- 校准器无需梯度训练,仅用 isotonic regression 拟合(sklearn 一行代码)
- ResNet-18 和 ResNet-50 两种 backbone
实验关键数据¶
主实验¶
| 测试场景 | 训练集→测试集 | Backbone | CPE (未校准) | CPE (校准后) | 改善幅度 | 角度误差 (未校准) | 角度误差 (校准后) |
|---|---|---|---|---|---|---|---|
| 跨被试 | MPII→MPII | ResNet18 | 23.17% | 5.18% | ↓78% | 5.77° | 5.09° (↓12%) |
| 跨被试 | RTGene→RTGene | ResNet18 | 19.60% | 5.26% | ↓73% | 12.36° | 10.55° (↓15%) |
| 跨数据集 | MPII→RTGene | ResNet18 | 20.60% | 4.75% | ↓77% | 13.71° | 10.12° (↓26%) |
| 跨数据集 | RTGene→MPII | ResNet18 | 27.21% | 4.84% | ↓82% | 18.46° | 14.50° (↓21%) |
| 跨数据集 | MPII→RTGene | ResNet50 | 20.10% | 4.63% | ↓77% | 13.89° | 9.50° (↓32%) |
所有场景 CPE 改善均超 70%,统计显著(Mann-Whitney U 检验 p<0.05)。
消融实验¶
| 校准样本数 | CPE (%) | 趋势说明 |
|---|---|---|
| 0 (未校准) | ~40% | 基线 |
| 10 | ~15% | 大幅改善 |
| 20 | ~10% | 持续改善 |
| 50 | ~5% | 接近饱和 |
| 100 | ~5% | 几乎无进一步改善 |
| 95% CI 覆盖率 | 分位回归 | 未校准模型 | 校准后模型 | 理想值 |
|---|---|---|---|---|
| Case 1 | 40.5% | 41.1% | 88.0% | 95% |
| Case 5 | 34.3% | 47.8% | 86.7% | 95% |
| Case 8 | 16.4% | 46.2% | 88.6% | 95% |
关键发现¶
- CPE 从 8-45% 范围一致降至 ~5%,稳定性极强
- 仅需 50 个校准样本即达饱和,数据效率远高于 meta-learning 等方法
- 校准还带来副产品——角度误差降低 7-32%,因为校准后的中位数比原始均值更鲁棒
- EUC 在校准前后几乎不变(始终 ~0.1-0.26),完全无法反映校准带来的真实改善,进一步证明 EUC 是不可靠指标
亮点与洞察¶
- 精准指出了视线追踪领域长期忽视的问题:不确定性估计在域偏移下严重失准,而社区一直在用有缺陷的 EUC 评价
- CPE 是一个通用的不确定性评估指标,不限于视线追踪领域
- 后验校准方法极度简洁(isotonic regression + ~50 样本),即插即用,不修改原模型
- 实验设计非常系统:4 种域偏移场景 × 2 种 backbone × 3 折交叉验证
局限性 / 可改进方向¶
- 校准假设校准集和测试集来自相同分布,当域偏移持续变化时(如从室内到室外)可能需要在线更新校准器
- 仅在 ResNet-18/50 + 高斯分布假设的模型上验证,未测试更复杂的模型架构(如 Transformer)或非参数分布
- MPIIGaze 和 RTGene 都是相对受控的数据集,更大域偏移(如极端光照、不同种族)的表现未知
- 校准器本身不提供不确定性估计,无法得知校准的可靠程度
- 未与贝叶斯神经网络、MC Dropout 等方法对比
与相关工作的对比¶
- Platt Scaling / Temperature Scaling:这些经典校准方法针对分类任务设计,假设 logit 到概率的映射是线性/仿射的。本文的 isotonic regression 校准器是非参数化的,对回归任务的非线性失准更具表达力,且不需要验证集上的梯度优化
- Quantile Regression:分位回归直接学习条件分位数,理论上可以避免分布假设。但实验显示分位回归在域偏移下同样严重失准(95% CI 覆盖率仅 16-40%),表明其学到的分位数也是域相关的。本文的后验校准方法可以叠加在分位回归之上进一步改善
- MC Dropout / Deep Ensembles:这些 epistemic 不确定性估计方法需要多次前向传播或训练多个模型,推理成本高。本文的方法仅需单次前向传播 + 后处理查表,实际部署代价几乎为零
- Meta-learning 域适应:如 FADA、MAML-based 方法可以在目标域微调不确定性估计,但需要较大的目标域标注数据和额外训练。本文仅需 ~50 个样本且无需梯度更新
启发与关联¶
- CPE 指标的通用性:CPE 不依赖于具体任务,可以直接用于评估任何输出概率分布的预测模型(如姿态估计、深度估计、天气预报)的不确定性质量。可以考虑在医学图像分割的不确定性评估中引入 CPE
- 后验校准的范式价值:"不修改模型,只修正输出分布" 这一思路对部署中的模型非常友好——模型已经上线,但发现在新场景下不确定性不准,只需收集少量新场景数据做校准即可
- 对视线追踪下游应用的影响:校准后的不确定性可以用于驾驶场景中的注意力分配决策——当视线估计不确定性高时自动切换到更保守的驾驶策略
评分¶
- 新颖性: ⭐⭐⭐⭐ CPE 指标和校准方法思路清晰,但技术贡献主要在组合而非开创
- 实验充分度: ⭐⭐⭐⭐⭐ 4 种域偏移场景+2 种backbone+交叉验证+样本量消融+case study,非常系统
- 写作质量: ⭐⭐⭐⭐ 逻辑清晰,图表直观,但部分符号略显冗余
- 价值: ⭐⭐⭐⭐ 方法实用性极强,CPE 指标对整个不确定性估计领域有参考价值