CoolFace
Datasetpublic

serteal/sparse-probing

Sparse Probing Datasets 155 binary classification tasks for probing language model representations. From: "Are Sparse Autoencoders Useful? A Case Study in Sparse Probing" (arXiv:2502.16681) Source: EleutherAI/sae-probes Usage from datasets import load_dataset # Load a specific dataset ds = load_dataset("serteal/sparse-probing", "87_glue_cola") # List available configurations from datasets import get_dataset_config_names configs =… See the full description on the dataset page: https://huggingface.co/datasets/serteal/sparse-probing.

sourceHugging Faceupdated 7mo agoView on Hugging Face
0likes452downloads
Dataset Card

Sparse Probing Datasets

155 binary classification tasks for probing language model representations.

From: "Are Sparse Autoencoders Useful? A Case Study in Sparse Probing" (arXiv:2502.16681)

Source: EleutherAI/sae-probes

Usage

python
from datasets import load_dataset

# Load a specific dataset
ds = load_dataset("serteal/sparse-probing", "87_glue_cola")

# List available configurations
from datasets import get_dataset_config_names
configs = get_dataset_config_names("serteal/sparse-probing")
print(f"Available: {len(configs)} datasets")

Dataset Categories

CategoryCountDescription
knowledge53Historical figures, geography, wikidata
reasoning27Science, ethics, commonsense, temporal
science30Domain concepts, medical, biology
moderation15Toxicity, spam, fake news, hate speech
glue10GLUE benchmark tasks
sentiment8Emotion, reviews
other9IT tickets, sports
code3Programming language detection

Available Datasets

Knowledge

Config NameDescription
2_hist_fig_birthyearBirth year of historical figures
3_hist_fig_deathyearDeath year of historical figures
4_hist_fig_ageAge of historical figures
5_hist_fig_ismaleGender classification of historical figures
6_hist_fig_isamericanAmerican nationality of historical figures
7_hist_fig_ispoliticianPolitician occupation of historical figures
8_nyc_latNYC latitude classification
9_nyc_longNYC longitude classification
10_nyc_boroughNYC borough classification
11_us_latUS latitude classification
12_us_longUS longitude classification
13_us_stateUS state classification
14_us_timezoneUS timezone classification
15_us_populationUS population classification
16_us_densityUS density classification
17_world_countryWorld country classification
18_world_latitudeWorld latitude classification
19_world_longitudeWorld longitude classification
20_world_pageviewsWorld entity pageviews classification
21_headline_istrumpHeadlines mentioning Trump
22_headline_isobamaHeadlines mentioning Obama
23_headline_ischinaHeadlines mentioning China
24_headline_isiranHeadlines mentioning Iran
25_headline_yearHeadline year classification
26_headline_isfrontpageFront page headline classification
27_art_typeArt type classification (book/song/movie)
28_art_yearArt year classification
29_art_pageviewsArt pageviews classification
30_book_lengthBook length classification
56_wikidatasex_or_genderWikidata gender classification
57_wikidatais_aliveWikidata alive status
58_wikidatapolitical_partyWikidata political party
59_wikidata_occupation_isjournalistWikidata journalist occupation
60_wikidata_occupation_isathleteWikidata athlete occupation
61_wikidata_occupation_isactorWikidata actor occupation
62_wikidata_occupation_ispoliticianWikidata politician occupation
63_wikidata_occupation_issingerWikidata singer occupation
64_wikidata_occupation_isresearcherWikidata researcher occupation
114_nyc_borough_ManhattanNYC Manhattan borough
115_nyc_borough_BrooklynNYC Brooklyn borough
116_nyc_borough_BronxNYC Bronx borough
117_us_state_FLUS Florida state
118_us_state_CAUS California state
119_us_state_TXUS Texas state
120_us_timezone_ChicagoUS Chicago timezone
121_us_timezone_New_YorkUS New York timezone
122_us_timezone_Los_AngelesUS Los Angeles timezone
123_world_country_United_KingdomWorld United Kingdom
124_world_country_United_StatesWorld United States
125_world_country_ItalyWorld Italy
126_art_type_bookArt type book classification
127_art_type_songArt type song classification
128_art_type_movieArt type movie classification

Reasoning

Config NameDescription
36_sciq_tfScience questions true/false
37_arith_mcArithmetic multiple choice
38_arith_difficultyArithmetic difficulty classification
39_arith_infixArithmetic infix notation
40_arith_rpnArithmetic reverse polish notation
41_truthqa_tfTruthfulQA true/false
42_temp_senseTemporal sense classification
43_temp_catTemporal category classification
44_phys_tfPhysical reasoning true/false
45_context_ansContextual answer classification
46_context_typeContext type classification
47_reasoning_tfCommonsense reasoning true/false
48_cm_correctCommon morality correctness
49_cm_isshortCommon morality statement brevity
50_deon_isvalidDeontological validity classification
51_just_isJustice classification
52_virtue_isVirtue ethics classification
54_cs_tfCommonsense true/false
55_open_qaOpen-domain QA classification
86_social_iqaSocial intelligence QA
129_arith_mc_AArithmetic multiple choice A
130_temp_cat_FrequencyTemporal category frequency
131_temp_cat_Typical TimeTemporal category typical time
132_temp_cat_Event OrderingTemporal category event ordering
133_context_type_CausalityContext type causality
134_context_type_Belief_statesContext type belief states
135_context_type_Event_durationContext type event duration

Science

Config NameDescription
65_high-schoolHigh school concept recognition
66_living-roomLiving room concept recognition
67_social-securitySocial security concept recognition
68_credit-cardCredit card concept recognition
69_blood-pressureBlood pressure concept recognition
70_prime-factorsPrime factors concept recognition
71_social-mediaSocial media concept recognition
72_gene-expressionGene expression concept recognition
73_control-groupControl group concept recognition
74_magnetic-fieldMagnetic field concept recognition
75_cell-linesCell lines concept recognition
76_trial-courtTrial court concept recognition
77_second-derivativeSecond derivative concept recognition
78_north-americaNorth America concept recognition
79_human-rightsHuman rights concept recognition
80_side-effectsSide effects concept recognition
81_public-healthPublic health concept recognition
82_federal-governmentFederal government concept recognition
83_third-partyThird party concept recognition
84_clinical-trialsClinical trials concept recognition
85_mental-healthMental health concept recognition
98_cancer_catCancer category classification
99_sci_catScience category classification
101_disease_classDisease classification
142_cancer_cat_Thyroid_CancerThyroid cancer classification
143_cancer_cat_Lung_CancerLung cancer classification
144_cancer_cat_Colon_CancerColon cancer classification
145_disease_class_digestive system diseasesDigestive system diseases
146_disease_class_cardiovascular diseasesCardiovascular diseases
147_disease_class_nervous system diseasesNervous system diseases

Moderation

Config NameDescription
94_ai_genAI-generated text detection
95_toxic_isToxicity detection
96_spam_isSpam detection
97_news_classNews category classification
100_news_fakeFake news detection
105_click_baitClickbait detection
106_hate_hateHate speech detection
107_hate_offensiveOffensive language detection
110_aimade_humangpt3Human vs GPT-3 text detection
139_news_class_PoliticsNews politics classification
140_news_class_TechnologyNews technology classification
141_news_class_EntertainmentNews entertainment classification
161_agnews_0AG News category 0
162_agnews_1AG News category 1
163_agnews_2AG News category 2

Glue

Config NameDescription
87_glue_colaGLUE CoLA linguistic acceptability
88_glue_mnliGLUE MNLI natural language inference
89_glue_mrpcGLUE MRPC paraphrase detection
90_glue_qnliGLUE QNLI question answering NLI
91_glue_qqpGLUE QQP Quora question pairs
92_glue_sst2GLUE SST2 sentiment analysis
93_glue_stsbGLUE STSB semantic similarity
136_glue_mnli_entailmentGLUE MNLI entailment
137_glue_mnli_neutralGLUE MNLI neutral
138_glue_mnli_contradictionGLUE MNLI contradiction

Sentiment

Config NameDescription
102_twt_emotionTweet emotion classification
111_yelp_dataYelp review sentiment
112_amzn_revAmazon review sentiment
113_movie_sentMovie review sentiment
148_twt_emotion_worryTweet emotion worry
149_twt_emotion_happinessTweet emotion happiness
150_twt_emotion_sadnessTweet emotion sadness
157_amazon_5starAmazon 5-star rating classification

Other

Config NameDescription
103_it_tickIT ticket classification
108_athlete_sportAthlete sport classification
109_ball_wsBasketball win shares classification
151_it_tick_HR SupportIT ticket HR support
152_it_tick_HardwareIT ticket hardware
153_it_tick_Administrative rightsIT ticket administrative rights
154_athlete_sport_footballAthlete football classification
155_athlete_sport_basketballAthlete basketball classification
156_athlete_sport_baseballAthlete baseball classification

Code

Config NameDescription
158_code_CC code detection
159_code_PythonPython code detection
160_code_HTMLHTML code detection

Schema

Each dataset contains:

  • prompt (string): The input text
  • label (int): Binary label (0 or 1)

Citation

bibtex
@article{makelov2025sparse,
  title={Are Sparse Autoencoders Useful? A Case Study in Sparse Probing},
  author={Makelov, Aleksandar and others},
  journal={arXiv preprint arXiv:2502.16681},
  year={2025}
}

License

Apache 2.0