Skip to main content
. Author manuscript; available in PMC: 2025 Jan 1.
Published in final edited form as: Proc Mach Learn Res. 2024 Jul;235:31344–31382.

Algorithm 1.

Robust multi-source confonnal prediction

1: Input: Training data 𝒟=𝒪i=Xi,Ti,Ri,RiYi,i=1,,n with number of sites K>0, and the target site is indexed by T=0; desired coverage probability 1-α; estimators of nuisance functions mk(θ,X), η0(X), and ωk,0(X) for k=1,,K-1; a tuning parameter λ (in the optimization step); a testing point X=x from the target site.
2: Output: A valid prediction set C^α(X).
3: Split the training data 𝒟 randomly into 𝒟1 and 𝒟2, where 𝒟j={𝒪i𝒟,ij} for j=1,2 and 12={1,2,,n}.
4: Fit nuisance functions m^k and ω^k,0 using SuperLearner on 𝒟1 and predict them on 𝒟2.
5: For the target site k=0, find θ^=r^0 that solves 0=12i2φ0(𝒪i;θ^,m^0,η^0).
6: For source sites k1, find θ^=r^k that solves 0=12i2φk(𝒪i;θ^,m^0,m^k,ω^k,0). Compute χ^k=r^0-r^k.
7: Solve for aggregation weights w^=w^0,w^1,w^K-1 that minimize Q(w) subject to 0wk1 and k=0K-1wk=1.
8: Compute θ^=r^0,fed=k=0K-1w^kr^k.
9: Return: The prediction set C^α(X)=y:S(x,y)r^0,fed.