Skip to content

Default to MultiTaskGP + LearnedFeatureImputation for heterogeneous TL (#5193)#5193

Open
hvarfner wants to merge 2 commits into
facebook:mainfrom
hvarfner:export-D102197137
Open

Default to MultiTaskGP + LearnedFeatureImputation for heterogeneous TL (#5193)#5193
hvarfner wants to merge 2 commits into
facebook:mainfrom
hvarfner:export-D102197137

Conversation

@hvarfner
Copy link
Copy Markdown

@hvarfner hvarfner commented Apr 28, 2026

Summary:

Switches the default heterogeneous transfer learning model from a specialized per-task kernel model to a standard multi-task GP with learned feature imputation. The previous default model class is marked as deprecated.

Reviewed By: saitcakmak

Differential Revision: D102197137

@meta-cla meta-cla Bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Apr 28, 2026
@meta-codesync
Copy link
Copy Markdown

meta-codesync Bot commented Apr 28, 2026

@hvarfner has exported this pull request. If you are a Meta employee, you can view the originating Diff in D102197137.

@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Apr 28, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 96.38%. Comparing base (47defa1) to head (4fcc258).

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #5193      +/-   ##
==========================================
- Coverage   96.38%   96.38%   -0.01%     
==========================================
  Files         617      617              
  Lines       69579    69605      +26     
==========================================
+ Hits        67065    67089      +24     
- Misses       2514     2516       +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

@meta-codesync meta-codesync Bot changed the title Default to MultiTaskGP + LearnedFeatureImputation for heterogeneous TL Default to MultiTaskGP + LearnedFeatureImputation for heterogeneous TL (#5193) Apr 29, 2026
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 29, 2026
facebook#5193)

Summary:

Switches the default heterogeneous transfer learning model from a specialized per-task kernel model to a standard multi-task GP with learned feature imputation. The previous default model class is marked as deprecated.

Differential Revision: D102197137
@hvarfner hvarfner force-pushed the export-D102197137 branch 2 times, most recently from 7e8ebf0 to aa4b964 Compare April 29, 2026 14:56
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 29, 2026
facebook#5193)

Summary:

Switches the default heterogeneous transfer learning model from a specialized per-task kernel model to a standard multi-task GP with learned feature imputation. The previous default model class is marked as deprecated.

Differential Revision: D102197137
Carl Hvarfner added 2 commits May 11, 2026 11:58
…TaskGP (facebook#5192)

Summary:
X-link: meta-pytorch/botorch#3296


Automatically configures learned feature imputation for models that pad heterogeneous per-task data to the full joint feature space. Models with native heterogeneity support are excluded from this automatic configuration.

Reviewed By: saitcakmak

Differential Revision: D101841497
facebook#5193)

Summary:

Switches the default heterogeneous transfer learning model from a specialized per-task kernel model to a standard multi-task GP with learned feature imputation. The previous default model class is marked as deprecated.

Reviewed By: saitcakmak

Differential Revision: D102197137
@hvarfner hvarfner force-pushed the export-D102197137 branch from aa4b964 to 4fcc258 Compare May 11, 2026 18:58
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request May 11, 2026
facebook#5193)

Summary:

Switches the default heterogeneous transfer learning model from a specialized per-task kernel model to a standard multi-task GP with learned feature imputation. The previous default model class is marked as deprecated.

Reviewed By: saitcakmak

Differential Revision: D102197137
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed Do not delete this pull request or issue due to inactivity. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants