1#!/usr/bin/env python32"""WP44 — Phase 6: emerging-research detection framework (RQ10, hypothesis H8), strictly out-of-time.34Units: OpenAlex topics (T-ids) restricted to those with >= 50 corpus works over 2018–2026 (topic-level series are the5most reproducible unit; semantic clusters are an extension in experiments/emerging_clusters.py).6Monthly panel from the eligible corpus (month-precision records only): n_{k,t} works per topic k and month t; citation7inflow c_{k,t} (internal citations received in month t by works of topic k, from cit_edges_internal with citing first8month); cross-area share x_{k,t}; mean novelty (optional, from embeddings) z_{k,t}.910Signals at cutoff T (use data <= T only; all means/sd computed on the trailing window ending at T):11 g12 growth of 12-month volume vs previous 12 months (log ratio)12 acc acceleration: g12 at T minus g12 at T-1213 zvol z-score of the last 3 months' volume against the trailing 24-month mean/sd14 cit_acc log ratio of citation inflow last 12 months vs previous 1215 xshare cross-area share of references in the last 12 months (interdisciplinary connectivity)16 share topic share of corpus volume (size control)17 age months since first appearance (>= 5 works)18Label at horizon h (12 or 24 months): emerging_{k,T,h} = 1 if the topic's growth g_h over (T, T+h] is in the top decile19among topics with >= 20 works in (T-12, T], AND its absolute volume in (T, T+h] >= 2x the volume in (T-h, T].20Models: z-score baseline (rank by zvol), logistic regression (L2) and gradient boosting trained on cutoffs21<= 2021-12 (labels observed by 2023-12) and tested on cutoffs 2023-01 … 2024-10 (labels observed by 2026-10).22Metrics: AUC, precision@20/50, lead time on the preregistered known-emergence set (protocol P1, H8).23Outputs: results/emerging/{panel.parquet, features.parquet, metrics.json, predictions.csv, known_set_lead_times.csv}.24"""25import argparse, json, os, duckdb, numpy as np, pandas as pd2627KNOWN_SET = { # preregistered (protocol P1, H8): OpenAlex topic ids mapped after taxonomy work; filled by experiments/known_set.csv28}293031def build_panel(root, con, last_month):32 con.execute(f"""CREATE OR REPLACE TABLE em_panel AS33 WITH W AS (SELECT c.id, c.first_month AS m, t.topic_id FROM corpus_works c JOIN oa_work_topics t ON t.work_id = c.id AND t.rank = 034 WHERE c.eligible AND c.date_precision = 'month' AND c.first_month BETWEEN '2018-01' AND '{last_month}'),35 V AS (SELECT topic_id, m, count(*) n FROM W GROUP BY 1,2),36 C AS (SELECT w.topic_id, cw.first_month AS m, count(*) c FROM cit_edges_internal e JOIN W w ON w.id = e.cited_id37 JOIN corpus_works cw ON cw.id = e.citing_id WHERE cw.date_precision = 'month' GROUP BY 1,2)38 SELECT v.topic_id, v.m, v.n, coalesce(c.c, 0) AS c FROM V v LEFT JOIN C c USING (topic_id, m)""")39 df = con.execute("SELECT * FROM em_panel").df()40 keep = df.groupby("topic_id")["n"].sum(); keep = keep[keep >= 50].index41 df = df[df.topic_id.isin(keep)]42 months = pd.period_range("2018-01", last_month, freq="M").strftime("%Y-%m")43 P = df.pivot_table(index="topic_id", columns="m", values="n", fill_value=0).reindex(columns=months, fill_value=0)44 C = df.pivot_table(index="topic_id", columns="m", values="c", fill_value=0).reindex(columns=months, fill_value=0)45 return P, C464748def features_at(P, C, T_idx):49 """Signals computed with columns <= T_idx only."""50 n = P.values; c = C.values51 def win(a, lo, hi): # sum over columns (lo, hi]52 lo = max(lo, -1); return a[:, lo + 1:hi + 1].sum(1)53 v12 = win(n, T_idx - 12, T_idx); v12p = win(n, T_idx - 24, T_idx - 12); v12pp = win(n, T_idx - 36, T_idx - 24)54 g12 = np.log1p(v12) - np.log1p(v12p); g12p = np.log1p(v12p) - np.log1p(v12pp)55 last3 = win(n, T_idx - 3, T_idx) / 3.056 trail = n[:, max(0, T_idx - 27):T_idx - 3 + 1]57 zvol = (last3 - trail.mean(1)) / (trail.std(1) + 1e-6)58 c12 = win(c, T_idx - 12, T_idx); c12p = win(c, T_idx - 24, T_idx - 12)59 cit_acc = np.log1p(c12) - np.log1p(c12p)60 cum = n[:, :T_idx + 1].cumsum(1); first = (cum >= 5).argmax(1); age = np.where(cum[:, -1] >= 5, T_idx - first, -1)61 share = v12 / max(v12.sum(), 1)62 return pd.DataFrame({"topic_id": P.index, "T_idx": T_idx, "v12": v12, "g12": g12, "acc": g12 - g12p, "zvol": zvol, "cit_acc": cit_acc,63 "share": share, "age": age, "c12": c12})646566def labels_at(P, T_idx, h):67 n = P.values68 pre = n[:, max(0, T_idx - h + 1):T_idx + 1].sum(1); post = n[:, T_idx + 1:T_idx + h + 1].sum(1)69 base12 = n[:, max(0, T_idx - 11):T_idx + 1].sum(1)70 g = np.log1p(post) - np.log1p(pre)71 elig = base12 >= 2072 thr = np.quantile(g[elig], 0.9) if elig.sum() > 10 else np.inf73 lab = ((g >= thr) & (post >= 2 * np.maximum(pre, 1)) & elig).astype(int)74 return pd.DataFrame({"topic_id": P.index, "T_idx": T_idx, "label": lab, "g_future": g, "eligible": elig})757677def main():78 ap = argparse.ArgumentParser(); ap.add_argument("--root", required=True); ap.add_argument("--last-month", default="2026-10")79 ap.add_argument("--horizon", type=int, default=12); a = ap.parse_args(); R = a.root80 out = os.path.join(R, "results/emerging"); os.makedirs(out, exist_ok=True)81 con = duckdb.connect(os.path.join(R, "data/processed/wp44.duckdb")); con.execute("SET threads TO 20; SET memory_limit='60GB';")82 P, C = build_panel(R, con, a.last_month); P.to_parquet(f"{out}/panel_volume.parquet"); C.to_parquet(f"{out}/panel_citations.parquet")83 months = list(P.columns); h = a.horizon84 def idx(m): return months.index(m)85 train_cuts = [idx(m) for m in months if "2019-12" <= m <= "2021-12" and idx(m) % 3 == 0]86 test_cuts = [idx(m) for m in months if "2023-01" <= m <= "2024-10" and idx(m) % 3 == 0 and idx(m) + h < len(months)]87 rows = []88 for T in train_cuts + test_cuts:89 f = features_at(P, C, T); l = labels_at(P, T, h); rows.append(f.merge(l, on=["topic_id", "T_idx"]))90 D = pd.concat(rows); D["split"] = np.where(D.T_idx.isin(train_cuts), "train", "test"); D = D[D.eligible]91 D.to_parquet(f"{out}/features.parquet")92 feats = ["g12", "acc", "zvol", "cit_acc", "share", "age"]93 from sklearn.linear_model import LogisticRegression94 from sklearn.ensemble import GradientBoostingClassifier95 from sklearn.metrics import roc_auc_score96 tr = D[D.split == "train"]; te = D[D.split == "test"]97 X = lambda d: d[feats].replace([np.inf, -np.inf], 0).fillna(0).values98 models = {"zscore_baseline": None, "logistic": LogisticRegression(C=1.0, max_iter=2000), "gbm": GradientBoostingClassifier(n_estimators=300, max_depth=3, learning_rate=0.05, random_state=44)}99 metrics = {"horizon": h, "n_train": int(len(tr)), "n_test": int(len(te)), "pos_rate_train": float(tr.label.mean()), "pos_rate_test": float(te.label.mean()),100 "train_cutoffs": [months[i] for i in train_cuts], "test_cutoffs": [months[i] for i in test_cuts]}101 preds = te[["topic_id", "T_idx", "label", "g_future"]].copy()102 for name, m in models.items():103 if m is None:104 s = te["zvol"].values105 else:106 m.fit(X(tr), tr.label); s = m.predict_proba(X(te))[:, 1]107 preds[name] = s108 res = {"auc": float(roc_auc_score(te.label, s))}109 for k in (20, 50):110 pk = []111 for T, g in preds.groupby("T_idx"):112 top = g.nlargest(k, name); pk.append(top.label.mean())113 res[f"precision_at_{k}"] = float(np.mean(pk))114 metrics[name] = res115 # per-cutoff AUC stability116 metrics["auc_by_cutoff"] = {months[T]: {n: float(roc_auc_score(g.label, g[n])) if g.label.nunique() > 1 else None for n in models} for T, g in preds.groupby("T_idx")}117 preds["month"] = [months[i] for i in preds.T_idx]; preds.to_csv(f"{out}/predictions.csv", index=False)118 json.dump(metrics, open(f"{out}/metrics_h{h}.json", "w"), indent=1); print(json.dumps({k: v for k, v in metrics.items() if k != "auc_by_cutoff"}, indent=1))119120121if __name__ == "__main__":122 main()123