Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix bug with Multitask DeepGP predictive variances. #2123

Merged
merged 3 commits into from
Sep 7, 2022

Conversation

gpleiss
Copy link
Member

@gpleiss gpleiss commented Sep 6, 2022

The latent variances of multitask DeepGP models are stored in non-interleaved covariance matrices.
Previously, the MultitaskMultivariateNormal.marginal method implicitly assumed that the function
covariance matrices were interleaved.

In particular, this affects multitask DeepGP models which use a
non-interleaved latent covariance matrix.

With this PR, MultitaskMultivariateNormal.marginal now checks if the input covariance matrix is
interleaved, and makes sure that the returned predictive covariance matrix matches the
interleaved/non-interleaved pattern of the latent covariance matrix.

[Fixes #2702]

The latent variances of multitask DeepGP models are stored in non-interleaved covariance matrices.
Previously, the MultitaskMultivariateNormal.marginal method implicitly assumed that the function
covariance matrices were interleaved.

In particular, this affects multitask DeepGP models which use a
non-interleaved latent covariance matrix.

With this PR, MultitaskMultivariateNormal.marginal now checks if the input covariance matrix is
interleaved, and makes sure that the returned predictive covariance matrix matches the
interleaved/non-interleaved pattern of the latent covariance matrix.

[Fixes #2702]
@gpleiss gpleiss added bug multitask For questions about multitask models labels Sep 6, 2022
Copy link
Collaborator

@Balandat Balandat left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm!

gpytorch/likelihoods/multitask_gaussian_likelihood.py Outdated Show resolved Hide resolved
@gpleiss
Copy link
Member Author

gpleiss commented Oct 3, 2022

Closed #2072 (I accidentally linked the wrong issue number in the original commit!)

@gpleiss gpleiss deleted the multitask_likelihood_non_interleaved branch October 3, 2022 20:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug multitask For questions about multitask models
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants