Skip to content

Use StratifiedStandardize for per-task Y standardization in TL (#5194)#5194

Open
hvarfner wants to merge 3 commits into
facebook:mainfrom
hvarfner:export-D102197139
Open

Use StratifiedStandardize for per-task Y standardization in TL (#5194)#5194
hvarfner wants to merge 3 commits into
facebook:mainfrom
hvarfner:export-D102197139

Conversation

@hvarfner
Copy link
Copy Markdown

@hvarfner hvarfner commented Apr 29, 2026

Summary:

Adds per-task outcome standardization to the transfer learning adapter, ensuring each task's observations are standardized independently rather than jointly. Updates the default transform pipeline to use TL-specific outcome transforms.

This removes ambiguity on whether the right transforms have been applied (e.g. QuickBO/warm-starting), where standardization is not performed across, but within experiments.

Differential Revision: D102197139

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

meta-codesync Bot commented Apr 29, 2026

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

Carl Hvarfner added 3 commits April 29, 2026 07:56
…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.

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.

Differential Revision: D102197137
…ook#5194)

Summary:

Adds per-task outcome standardization to the transfer learning adapter, ensuring each task's observations are standardized independently rather than jointly. Updates the default transform pipeline to use TL-specific outcome transforms.

This removes ambiguity on whether the right transforms have been applied (e.g. QuickBO/warm-starting), where standardization is not performed across, but within experiments.

Differential Revision: D102197139
@hvarfner hvarfner force-pushed the export-D102197139 branch from 8046f90 to 407478c Compare April 29, 2026 14:56
@meta-codesync meta-codesync Bot changed the title Use StratifiedStandardize for per-task Y standardization in TL Use StratifiedStandardize for per-task Y standardization in TL (#5194) Apr 29, 2026
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 29, 2026
…ook#5194)

Summary:

Adds per-task outcome standardization to the transfer learning adapter, ensuring each task's observations are standardized independently rather than jointly. Updates the default transform pipeline to use TL-specific outcome transforms.

This removes ambiguity on whether the right transforms have been applied (e.g. QuickBO/warm-starting), where standardization is not performed across, but within experiments.

Differential Revision: D102197139
@codecov-commenter
Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 98.30508% with 1 line in your changes missing coverage. Please review.
✅ Project coverage is 96.38%. Comparing base (12ebbd9) to head (407478c).

Files with missing lines Patch % Lines
ax/adapter/transfer_learning/adapter.py 50.00% 1 Missing ⚠️
Additional details and impacted files
@@           Coverage Diff           @@
##             main    #5194   +/-   ##
=======================================
  Coverage   96.38%   96.38%           
=======================================
  Files         617      617           
  Lines       69463    69491   +28     
=======================================
+ Hits        66954    66981   +27     
- Misses       2509     2510    +1     

☔ 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.

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