CoolFace
Datasetpublic

taylor-geospatial/MINDSET

MINDSET MINDSET is the pretraining dataset for MIND, a coordinate-only location encoder distilled from static location encoder teachers and annual AlphaEarth Foundations (AEF) embeddings. We release the embeddings at the 12.1M training coordinates. The dataset contains 12,099,072 land coordinates in WGS84. Coordinates are dense around cities and not uniformly sampled over land. The files are in GeoParquet format and can be joined on point_id: file grain rows columns… See the full description on the dataset page: https://huggingface.co/datasets/taylor-geospatial/MINDSET.

sourceHugging Facecc-by-4.0updated 3d agoView on Hugging Face
0likes359downloads
Dataset Card

MINDSET

MINDSET is the pretraining dataset for MIND, a coordinate-only location encoder distilled from static location encoder teachers and annual AlphaEarth Foundations (AEF) embeddings.

We release the embeddings at the 12.1M training coordinates. The dataset contains 12,099,072 land coordinates in WGS84. Coordinates are dense around cities and not uniformly sampled over land.

The files are in GeoParquet format and can be joined on point_id:

filegrainrowscolumns
mindset_teachers.parquetone row per point12,099,072point_id, geometry, bbox, climplicit [1024], geoclip [512], sinr [256]
mindset_aef.parquetone row per point and year108,891,648point_id, year, geometry, bbox, aef [64]

Coordinate and array conventions

  • Geometry is OGC:CRS84: Point(lon, lat) in WGS84 degrees.
  • The coordinate input used by MIND is ordered (lat, lon); the GeoParquet geometry is ordered (lon, lat).
  • The static teacher arrays are raw float16 embedder outputs. L2-normalize them when constructing targets.
  • AEF is stored as native signed int8. For valid values, dequantize with f = sign(x) * (|x| / 127.5) ** 2; -128 is nodata. L2-normalize after dequantization.
  • AEF covers years 2017--2025.

Teachers

columnteacherdimensionsource
aefAlphaEarth Foundations64Google AEF v1 annual, source.coop tge-labs/aef
climplicitClimplicit1024Jobedo/climplicit
geoclipGeoCLIP512GeoCLIP location encoder
sinrSINR256MVRL/sinr-location-encoder-1000-cls

The dataset contains CC-BY-4.0 metadata. The files contain embeddings derived from upstream models, so the license and use terms of each upstream model and source apply.

Load

python
import pyarrow.dataset as ds
import huggingface_hub
import fsspec

teachers = ds.dataset(
    "hf://datasets/XXXX/MINDSET/mindset_teachers.parquet", format="parquet"
)
aef = ds.dataset(
    "hf://datasets/XXXX/MINDSET/mindset_aef.parquet", format="parquet"
)
# Scan only columns and rows needed for a spatial AOI
sample = teachers.to_table(columns=["point_id", "geometry"], filter=(ds.field("point_id") < 10))