Hmm... if documents are just an ID and a list of token IDs, maybe something like a SparseVector is a better representation?