大型预训练模型在自然语言处理和计算机视觉等领域取得了巨大成功,但对其进行全参数微调的计算成本极高。低秩适配(LoRA)作为一种高效的微调方法,通过引入低秩矩阵来近似权重更新,显著降低了计算开销,成为事实上的标准。然而,LoRA也存在固有局限:其低秩点估计更新限制了模型的表达能力,导致与全微调之间存在持续的性能差距;同时,它缺乏内置的不确定性量化能力,这在可靠性至关重要的应用场景中是一个严重缺陷。
为了解决这些问题,研究人员提出了BaLoRA,即贝叶斯低秩适配。BaLoRA是LoRA的贝叶斯扩展,其核心创新在于一种输入自适应的贝叶斯参数化方法。该方法对LoRA矩阵施加贝叶斯先验,并通过后验推断实现自适应噪声注入。令人惊讶的是,这种贝叶斯扩展不仅提供了校准良好的不确定性估计,更重要的是,自适应噪声注入显著提升了预测精度,在自然语言推理和视觉任务中均缩小了与全微调之间的差距。
为了验证BaLoRA的实际效果,研究团队将其应用于金属有机框架(MOF)的带隙预测任务。MOF是一种具有广泛应用前景的多孔材料,其带隙预测对于材料设计至关重要。实验结果表明,BaLoRA能够生成零样本的测试时不确定性估计,这些估计与模型误差的相关性比经过训练的LoRA集成更强。此外,随着计算资源的增加,BaLoRA的不确定性估计单调改善,且精度不受影响。
BaLoRA的提出为大规模模型的微调提供了一种新的思路。它不仅提升了模型在关键任务中的可靠性,还通过贝叶斯框架引入了不确定性量化能力,使得模型在科学发现、医疗诊断等高风险管理领域更加可信。未来,BaLoRA有望成为LoRA的替代方案,在更多实际应用中发挥重要作用。