Skip to content

Upper Limit for training data in Exact GPs #1928

Answered by wjmaddox
jejjohnson asked this question in Q&A
Discussion options

You must be logged in to vote

With KeOps, I've actually run the full house electric dataset (1.8ish million data points) on a single RTX GPU and that takes roughly 12 hours for 50 optimization steps using Adam. I haven't tried going beyond that recently (but I think it's possible if you're willing to run something for 1-2 days) or trying to use KeOps plus kernel checkpointing.

The falkon paper also uses KeOps but for kernel ridge regression and seems to get into the billions of data points with Nystrom approximations (so maybe not really exact then).

It's possible other people have had different experiences / more detailed knowledge though.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by jejjohnson
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants