首页
/ GPyTorch v1.14版本发布:全面支持Python 3.10与PyTorch 2.0

GPyTorch v1.14版本发布:全面支持Python 3.10与PyTorch 2.0

2025-06-14 13:51:14作者:幸俭卉

GPyTorch是基于PyTorch的高斯过程(Gaussian Process)机器学习库,它提供了构建和训练高斯过程模型的灵活框架。高斯过程是一种强大的非参数贝叶斯方法,广泛应用于回归、分类、优化等领域。GPyTorch通过利用PyTorch的自动微分和GPU加速能力,使得高斯过程模型能够高效地处理大规模数据。

核心改进与功能增强

线性核函数维度特异性方差支持

新版本对线性核函数进行了重要扩展,现在支持为每个输入维度设置独立的方差参数。这一改进使得模型能够更好地捕捉不同特征的重要性差异,在特征重要性分析任务中尤为有用。开发者可以通过设置variance_per_dimension=True参数来启用这一功能。

Softmax似然函数权重参数优化

修复了SoftmaxLikelihood中一个长期存在的问题:即使明确设置mixing_weights=False,模型仍会添加可学习的权重参数W。这一修复确保了模型配置的精确性,避免了不必要的参数计算,提高了模型效率。

多维正态分布操作增强

新版本对多维正态分布(MVN)类进行了多项功能增强:

  • 实现了unsqueeze方法,支持在指定维度扩展张量
  • 改进了expand方法,现在支持非惰性(non-lazy)多维正态分布
  • 优化了协方差矩阵的复用机制,减少了不必要的计算

这些改进使得批量处理和多任务学习场景下的操作更加灵活高效。

类型注解与代码质量提升

v1.14版本在代码类型注解方面取得了显著进展:

  • gpytorch.Module基类添加了全面的类型注解
  • 改进了线性核函数的类型提示
  • 开始为ExactGP类添加类型注解

这些改进不仅提升了代码的可读性和可维护性,还为开发者提供了更好的IDE支持,减少了类型相关的错误。

性能优化与错误修复

  • 修复了KeOps核矩阵对角线计算的问题,确保了数值稳定性
  • 改进了梯度计算块的符号处理,提高了优化过程的准确性
  • 优化了幻想(fantasization)后的JIT追踪支持,通过分离new_covar_cache实现了更好的模型序列化

向后兼容性说明

值得注意的是,v1.14版本移除了已被弃用的惰性张量(LazyTensor)实现,开发者需要确保代码中不再使用这些已被移除的功能。同时,该版本全面支持Python 3.10和PyTorch 2.0,建议用户升级到这些较新的版本以获得最佳体验。

应用场景与影响

GPyTorch v1.14的这些改进特别有利于以下应用场景:

  1. 高维特征选择任务:通过维度特异性方差支持,可以更精确地评估各特征的重要性
  2. 多任务学习:增强的多维正态分布操作简化了复杂模型的构建
  3. 生产环境部署:改进的类型系统和JIT支持提高了代码的可靠性

对于高斯过程研究者和实践者来说,这一版本标志着GPyTorch在功能性、稳定性和开发者体验方面的重要进步,为构建更复杂、更高效的高斯过程模型奠定了坚实基础。

登录后查看全文
热门项目推荐

项目优选

收起
openHiTLS-examplesopenHiTLS-examples
本仓将为广大高校开发者提供开源实践和创新开发平台,收集和展示openHiTLS示例代码及创新应用,欢迎大家投稿,让全世界看到您的精巧密码实现设计,也让更多人通过您的优秀成果,理解、喜爱上密码技术。
C
53
468
kernelkernel
deepin linux kernel
C
22
5
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
349
381
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
7
0
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
133
186
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
878
517
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
336
1.1 K
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
180
264
cherry-studiocherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
612
60
note-gennote-gen
一款跨平台的 Markdown AI 笔记软件,致力于使用 AI 建立记录和写作的桥梁。
TSX
83
4