2025-11-11T12:13:09.699032

Budget-constrained Active Learning to Effectively De-censor Survival Data

Parsaee, Jiang, Friggstad et al.

Standard supervised learners attempt to learn a model from a labeled dataset. Given a small set of labeled instances, and a pool of unlabeled instances, a budgeted learner can use its given budget to pay to acquire the labels of some unlabeled instances, which it can then use to produce a model. Here, we explore budgeted learning in the context of survival datasets, which include (right) censored instances, where we know only a lower bound on an instance's time-to-event. Here, that learner can pay to (partially) label a censored instance -- e.g., to acquire the actual time for an instance [perhaps go from (3 yr, censored) to (7.2 yr, uncensored)], or other variants [e.g., learn about one more year, so go from (3 yr, censored) to either (4 yr, censored) or perhaps (3.2 yr, uncensored)]. This serves as a model of real world data collection, where follow-up with censored patients does not always lead to uncensoring, and how much information is given to the learner model during data collection is a function of the budget and the nature of the data itself. We provide both experimental and theoretical results for how to apply state-of-the-art budgeted learning algorithms to survival data and the respective limitations that exist in doing so. Our approach provides bounds and time complexity asymptotically equivalent to the standard active learning method BatchBALD. Moreover, empirical analysis on several survival tasks show that our model performs better than other potential approaches on several benchmarks.

academic

বাজেট-সীমাবদ্ধ সক্রিয় শিক্ষা দ্বারা সার্ভাইভাল ডেটা কার্যকরভাবে ডি-সেন্সর করা

মৌলিক তথ্য

পেপার আইডি: 2510.12144
শিরোনাম: Budget-constrained Active Learning to Effectively De-censor Survival Data
লেখক: Ali Parsaee, Bei Jiang, Zachary Friggstad, Russell Greiner (University of Alberta)
শ্রেণীবিভাগ: cs.LG cs.AI
প্রকাশনা সময়: অক্টোবর ১৫, ২০২৫
পেপার লিংক: https://arxiv.org/abs/2510.12144

সারসংক্ষেপ

এই পেপারটি সার্ভাইভাল ডেটাসেটে বাজেট-সীমাবদ্ধ সক্রিয় শিক্ষার সমস্যা অন্বেষণ করে। সার্ভাইভাল ডেটায় ডান-সেন্সর করা উদাহরণ রয়েছে, যেখানে আমরা শুধুমাত্র ইভেন্ট ঘটার সময়ের নিম্ন সীমা জানি। শিক্ষার্থী বাজেট ব্যয় করে সেন্সর করা উদাহরণগুলি (আংশিকভাবে) লেবেল করতে পারে, যেমন "(৩ বছর, সেন্সর করা)" থেকে প্রকৃত সময় "(৭.২ বছর, সেন্সর করা নয়)" পেতে পারে, অথবা অন্যান্য ভেরিয়েন্ট যেমন "(৩ বছর, সেন্সর করা)" থেকে "(৪ বছর, সেন্সর করা)" বা "(৩.২ বছর, সেন্সর করা নয়)"। এটি বাস্তব-বিশ্বের ডেটা সংগ্রহ প্রক্রিয়া অনুকরণ করে, যেখানে সেন্সর করা রোগীদের ফলো-আপ সর্বদা ডি-সেন্সরিং এর দিকে পরিচালিত করে না, এবং শিক্ষার্থীর মডেল ডেটা সংগ্রহ প্রক্রিয়ায় যে তথ্য পায় তা বাজেট এবং ডেটার প্রকৃতির একটি ফাংশন।

গবেষণা পটভূমি এবং প্রেরণা

সমস্যা সংজ্ঞা

১. মূল সমস্যা: বাজেট সীমাবদ্ধতার অধীনে, সার্ভাইভাল পূর্বাভাস মডেলের কর্মক্ষমতা সর্বাধিক করতে কোন সেন্সর করা উদাহরণগুলি ডি-সেন্সর করার জন্য কার্যকরভাবে নির্বাচন করবেন ২. ব্যবহারিক তাৎপর্য:

চিকিৎসা গবেষণায় রোগী ফলো-আপের উচ্চ খরচ
শিল্প নির্ভরযোগ্যতা পরীক্ষায় অতিরিক্ত পরীক্ষার খরচ
অ্যালগরিদম রানটাইম পূর্বাভাসে গণনা খরচ

বিদ্যমান পদ্ধতির সীমাবদ্ধতা

১. ঐতিহ্যবাহী সক্রিয় শিক্ষা: প্রধানত শ্রেণীবিভাগ এবং রিগ্রেশন কাজের জন্য, সেন্সর করা ডেটার বিশেষত্ব বিবেচনা করে না २. সার্ভাইভাল বিশ্লেষণে সক্রিয় শিক্ষা: গবেষণা বিরল, বাজেট সীমাবদ্ধতা বিবেচনার অভাব ३. BatchBALD সীমাবদ্ধতা:

ধরে নেয় যে oracle সম্পূর্ণ লেবেল তথ্য প্রদান করে
পৃথক উদাহরণের বিভিন্ন খরচ বিবেচনা করে না
আংশিক ডি-সেন্সরিং পরিস্থিতিতে প্রযোজ্য নয়

গবেষণা প্রেরণা

বাস্তব-বিশ্বে ডেটা সংগ্রহের খরচ বেশি, বিশেষত চিকিৎসা গবেষণা, শিল্প পরীক্ষা এবং অন্যান্য ক্ষেত্রে। ঐতিহ্যবাহী পদ্ধতিগুলি বাজেট সীমাবদ্ধতা এবং সেন্সর করা ডেটার বিশেষত্ব উপেক্ষা করে, এই জটিল পরিস্থিতি পরিচালনার জন্য বিশেষায়িত পদ্ধতির প্রয়োজন।

মূল অবদান

१. আনুষ্ঠানিক সংজ্ঞা: বাজেট সীমাবদ্ধতার অধীনে সেন্সর করা উদাহরণগুলি ডি-সেন্সর করার শিক্ষা সমস্যা প্রথমবারের মতো আনুষ্ঠানিকভাবে সংজ্ঞায়িত করা २. অ্যালগরিদম উদ্ভাবন: BBsurv অ্যালগরিদম প্রস্তাব করা, যা সার্ভাইভাল ডেটা এবং বিভিন্ন উদাহরণ খরচ পরিচালনা করতে BatchBALD সামঞ্জস্য করে ३. তাত্ত্বিক গ্যারান্টি: প্রমাণ করা যে অ্যালগরিদম বহুপদী সময়ে সর্বোত্তম নিম্ন সীমা (১-१/e) অর্জন করে ४. ব্যাপক মূল্যায়ন: তিনটি প্রকৃত সার্ভাইভাল ডেটাসেটে ব্যাপক পরীক্ষা, পদ্ধতির শক্তিশালীতা প্রদর্শন করা ५. বেঞ্চমার্ক প্রতিষ্ঠা: আটটি তুলনামূলক অ্যালগরিদম প্রদান করা, এই কাজের জন্য মূল্যায়ন বেঞ্চমার্ক প্রতিষ্ঠা করা

পদ্ধতি বিস্তারিত

কাজের সংজ্ঞা

ইনপুট:

প্রোব গভীরতা k ∈ ℜ+ (প্রতিটি প্রোব অন্বেষণ করা বছরের সংখ্যা)
বাজেট B ∈ ℜ+
প্রশিক্ষণ ডেটাসেট D = {xi, ti, δi, ci}Li=1, যেখানে:
- xi: সহ-পরিবর্তনশীল
- ti: সময়
- δi: সেন্সরিং ফ্ল্যাগ (১ অসেন্সর করা, ० সেন্সর করা)
- ci: প্রোব খরচ

আউটপুট: উদাহরণ সেট F নির্বাচন করা যাতে ∑j∈F cj ≤ B, মডেল কর্মক্ষমতা সর্বাধিক করা

মডেল আর্কিটেকচার

१. বেয়েসিয়ান সার্ভাইভাল মডেল

বেয়েসিয়ান মাল্টি-টাস্ক লজিস্টিক রিগ্রেশন (MTLR) মডেল ব্যবহার করা:

ক্রমাগত সময়কে n সময় ব্যবধানে বিচ্ছিন্ন করা {bi}ni=1
মাল্টিনোমিয়াল বিতরণ আউটপুট {p(y = bi|x, ω, D)}ni=1
ব্যক্তিগত সার্ভাইভাল বিতরণ (ISD) উৎপন্ন করা

२. BBsurv অ্যালগরিদম মূল

সম্ভাব্যতা সমন্বয় প্রক্রিয়া:

pcens(y = bi|ω) = p(y = bi|ω) / ∑nr=i p(y = br|ω)

পরিচিত ব্যবধান প্রক্রিয়াকরণ:

প্রোব গভীরতা k এর মধ্যে "পরিচিত" ব্যবধান চিহ্নিত করা
প্রোব পরিসীমা অতিক্রম করা ব্যবধানগুলি একটি একক "অজানা" শ্রেণী buk এ একত্রিত করা
চূড়ান্ত সম্ভাব্যতা বিতরণ pfinal উৎপন্ন করা

३. অধিগ্রহণ ফাংশন

BatchBALD-ভিত্তিক পারস্পরিক তথ্য গণনা:

I(y1:b; ω|x1:b, D) = H(y1:b|x1:b, D) - Ep(ω|D,x1:b)[H(y1:b|x1:b, ω, D)]

প্রযুক্তিগত উদ্ভাবন পয়েন্ট

१. প্রোব গভীরতা মডেলিং: আংশিক ডি-সেন্সরিং প্রোব গভীরতা ধারণায় উদ্ভাবনীভাবে মডেল করা २. সম্ভাব্যতা পুনর্বন্টন: সেন্সরিং সময়ের আগে শূন্য সম্ভাব্যতা ব্যবধান চতুরভাবে পরিচালনা করা ३. বাজেট অপ্টিমাইজেশন: সমস্যাটি ওজনযুক্ত সর্বাধিক কভারেজ সমস্যায় হ্রাস করা, লোভী অ্যালগরিদম ব্যবহার করে সমাধান করা ४. একীভূত কাঠামো: একযোগে অভিন্ন এবং অ-অভিন্ন খরচ সেটিং পরিচালনা করা

পরীক্ষা সেটআপ

ডেটাসেট

१. MIMIC-IV: ३८,५२० রোগী, ९३ বৈশিষ্ট্য, ६७% সেন্সরিং হার २. NACD: २,४०२ রোগী, ५३ বৈশিষ্ট্য, ३६% সেন্সরিং হার
३. SUPPORT: ९,१०५ রোগী, ४२ বৈশিষ্ট্য, ३२% সেন্সরিং হার

মূল্যায়ন মেট্রিক্স

প্রধান মেট্রিক: MAE-PO (সিউডো অবজারভেশন সহ গড় পরম ত্রুটি)
সহায়ক মেট্রিক্স: C-index, একীভূত Brier স্কোর, অসেন্সর করা ডেটা MAE

তুলনামূলক পদ্ধতি

१. BatchBALD: মূল BatchBALD অ্যালগরিদম २. C-BALD: সেন্সরিং-সচেতন BALD ভেরিয়েন্ট ३. IDEAL: বিপরীত দূরত্ব-ওজনযুক্ত সক্রিয় শিক্ষা ४. Entropy Sampling: এন্ট্রপি নমুনা ५. Variance Sampling: বৈচিত্র্য নমুনা ६. Closest to Half (CtH): ०.५ সম্ভাব্যতার কাছাকাছি নমুনা ७. Mean Closest to Middle (MCtM): গড় মধ্যবিন্দু নমুনা ८. Clusters to form Batches (CfB): ব্যাচ গঠনের জন্য ক্লাস্টারিং ९. Random: র্যান্ডম নমুনা

বাস্তবায়ন বিবরণ

১০টি সময় ব্যবধান ব্যবহার করা (কোয়ান্টাইল বিভাজনের উপর ভিত্তি করে)
Spike-and-Slab পূর্ব সহ বেয়েসিয়ান MTLR মডেল
৫০০০ প্রশিক্ষণ রাউন্ড
অ-তথ্যপূর্ণ সেন্সরিং নিশ্চিত করতে কৃত্রিম সেন্সরিং

পরীক্ষার ফলাফল

প্রধান ফলাফল

টেবিল १ বাজেট=१० এ MAE-PO ফলাফল দেখায়:

BBsurv বেশিরভাগ সেটিংয়ে অন্যান্য পদ্ধতির চেয়ে উল্লেখযোগ্যভাবে ভাল
প্রোব গভীরতা বৃদ্ধির সাথে, BBsurv এবং BatchBALD কর্মক্ষমতা একত্রিত হতে থাকে
MIMIC ডেটাসেটে, BBsurv BatchBALD এর তুলনায় সবচেয়ে উল্লেখযোগ্য উন্নতি

মূল আবিষ্কার: १. প্রোব গভীরতার প্রভাব: k=५ এ BBsurv সুবিধা সর্বাধিক, k=१०० এ BatchBALD এর কাছাকাছি २. ডেটাসেট পার্থক্য: MIMIC এবং NACD-তে উল্লেখযোগ্য উন্নতি, SUPPORT-এ ছোট পার্থক্য ३. পরিসংখ্যানগত তাৎপর্য: বেশিরভাগ ক্ষেত্রে p<०.०५ তাৎপর্য স্তর অর্জন করা

বাজেট সংবেদনশীলতা বিশ্লেষণ

চিত্র २ ক্রস-বাজেট কর্মক্ষমতা দেখায়:

অভিন্ন খরচ সেটিং: BBsurv সমস্ত বাজেট স্তরে ধারাবাহিকভাবে সর্বোত্তম
অ-অভিন্ন খরচ সেটিং: BBsurv সুবিধা আরও স্পষ্ট, বিশেষত উচ্চ বাজেটে
খরচ প্রক্রিয়াকরণ সুবিধা: পারস্পরিক তথ্যের সাব-মডুলারিটি BBsurv কে বাজেট সীমাবদ্ধতা আরও ভালভাবে পরিচালনা করতে সক্ষম করে

অ্যাবলেশন পরীক্ষা

প্রোব গভীরতার প্রভাব:

k=५: BBsurv বেসলাইনের চেয়ে উল্লেখযোগ্যভাবে ভাল
k=१०: মধ্যম উন্নতি
k=१००: BatchBALD কর্মক্ষমতার কাছাকাছি

খরচ সেটিং তুলনা:

অভিন্ন খরচ: বেশিরভাগ পদ্ধতির কর্মক্ষমতা একই রকম
অ-অভিন্ন খরচ: BBsurv এবং BatchBALD অন্যান্য পদ্ধতির চেয়ে উল্লেখযোগ্যভাবে ভাল

পরীক্ষামূলক আবিষ্কার

१. বৈচিত্র্যময় নির্বাচন: PCA ভিজুয়ালাইজেশন দেখায় যে BBsurv আরও বৈচিত্র্যময় উদাহরণ নির্বাচন করে २. CfB অপ্রত্যাশিত কর্মক্ষমতা: ক্লাস্টারিং পদ্ধতি কিছু সেটিংয়ে চমৎকার কর্মক্ষমতা প্রদর্শন করে ३. খরচ সংবেদনশীলতা: অ-অভিন্ন খরচ সেটিংয়ে, পারস্পরিক তথ্য-ভিত্তিক পদ্ধতির সুবিধা আরও স্পষ্ট

উপসংহার এবং আলোচনা

প্রধান উপসংহার

१. পদ্ধতির কার্যকারিতা: BBsurv বেশিরভাগ সেটিংয়ে বিদ্যমান পদ্ধতির চেয়ে ভাল २. তাত্ত্বিক গ্যারান্টি: অ্যালগরিদম জটিলতা BatchBALD এর সমান, একই সাথে সর্বোত্তম আনুমানিক অনুপাত প্রদান করে ३. ব্যবহারিক মূল্য: চিকিৎসা গবেষণা, শিল্প পরীক্ষা ইত্যাদি বাস্তব পরিস্থিতিতে প্রযোজ্য ४. শক্তিশালীতা: বিভিন্ন ডেটাসেট, বাজেট, প্রোব গভীরতায় স্থিতিশীল কর্মক্ষমতা

সীমাবদ্ধতা

१. অ-তথ্যপূর্ণ সেন্সরিং অনুমান: বাস্তব প্রয়োগে হতে পারে না २. নির্দিষ্ট প্রোব গভীরতা: প্রোব গভীরতা গতিশীল সমন্বয় বিবেচনা করে না ३. বিচ্ছিন্নকরণ আনুমানিক: সময় বিচ্ছিন্নকরণ তথ্য হারাতে পারে ४. গণনা জটিলতা: লোভী অ্যালগরিদম বড় আকারের ডেটায় ধীর হতে পারে

ভবিষ্যত দিকনির্দেশনা

१. আধা-তত্ত্বাবধানী সম্প্রসারণ: অসেন্সর করা ডেটা সহ কর্মক্ষমতা উন্নত করা २. তথ্যপূর্ণ সেন্সরিং: অ-তথ্যপূর্ণ সেন্সরিং অনুমান শিথিল করা ३. গতিশীল প্রোব: উদাহরণ বৈশিষ্ট্যের উপর ভিত্তি করে প্রোব গভীরতা সমন্বয় করা ४. আনুমানিক অ্যালগরিদম: আরও দক্ষ সর্বাধিক কভারেজ আনুমানিক স্কিম অন্বেষণ করা

গভীর মূল্যায়ন

শক্তি

१. সমস্যা উদ্ভাবনী: বাজেট সীমাবদ্ধতার অধীনে সার্ভাইভাল ডেটা ডি-সেন্সরিং সমস্যা প্রথমবারের মতো সিস্টেমেটিকভাবে গবেষণা করা २. পদ্ধতি কঠোরতা:

সম্পূর্ণ তাত্ত্বিক বিশ্লেষণ, জটিলতা এবং আনুমানিক অনুপাত গ্যারান্টি প্রদান করা
চতুর অ্যালগরিদম ডিজাইন, আংশিক তথ্য অধিগ্রহণ কার্যকরভাবে পরিচালনা করা ३. পরীক্ষা সম্পূর্ণতা:
তিনটি প্রকৃত ডেটাসেট, একাধিক মূল্যায়ন মেট্রিক্স
ব্যাপক বেসলাইন তুলনা এবং অ্যাবলেশন পরীক্ষা
পরিসংখ্যানগত তাৎপর্য যাচাইকরণ ४. ব্যবহারিক মূল্য উচ্চ: চিকিৎসা, শিল্প ইত্যাদি ক্ষেত্রের বাস্তব চাহিদা সমাধান করা

অপূর্ণতা

१. অনুমান সীমাবদ্ধতা: অ-তথ্যপূর্ণ সেন্সরিং অনুমান বাস্তবে হতে পারে না २. পদ্ধতি সীমাবদ্ধতা:

বিচ্ছিন্নকরণ প্রক্রিয়াকরণ ক্রমাগত সময় তথ্য হারাতে পারে
নির্দিষ্ট প্রোব গভীরতা নমনীয়তার অভাব ३. পরীক্ষা পরিসীমা:
ডেটাসেট আকার তুলনামূলকভাবে সীমিত
আরও SOTA সার্ভাইভাল বিশ্লেষণ পদ্ধতির সাথে তুলনার অভাব ४. তাত্ত্বিক বিশ্লেষণ: সংমিশ্রণ এবং সাধারণীকরণ ত্রুটি বিশ্লেষণ প্রদান করা হয়নি

প্রভাব

१. একাডেমিক অবদান:

নতুন গবেষণা দিকনির্দেশনা খোলা, পরবর্তী কাজ প্রত্যাশিত
তাত্ত্বিক কাঠামো অন্যান্য অসম্পূর্ণ তথ্য শিক্ষা সমস্যায় প্রসারিত করা যায় २. ব্যবহারিক মূল্য:
ক্লিনিকাল ট্রায়াল ডিজাইনে সরাসরি প্রয়োগ
শিল্প গুণমান নিয়ন্ত্রণ এবং নির্ভরযোগ্যতা পরীক্ষায় ব্যবহারযোগ্য ३. পদ্ধতি সর্বজনীনতা: কাঠামো অন্যান্য সক্রিয় শিক্ষা অ্যালগরিদমে অভিযোজিত হতে পারে

প্রযোজ্য পরিস্থিতি

१. চিকিৎসা গবেষণা: রোগী ফলো-আপ, ক্লিনিকাল ট্রায়াল ডিজাইন २. শিল্প প্রয়োগ: পণ্য জীবনকাল পরীক্ষা, ব্যর্থতা পূর্বাভাস ३. অ্যালগরিদম বিশ্লেষণ: রানটাইম পূর্বাভাস, কর্মক্ষমতা মূল্যায়ন ४. আর্থিক ক্ষেত্র: ক্রেডিট ঝুঁকি মূল্যায়ন, ডিফল্ট পূর্বাভাস

রেফারেন্স

পেপারটি ४१টি সম্পর্কিত সাহিত্য উদ্ধৃত করে, প্রধানত অন্তর্ভুক্ত:

BatchBALD মূল পেপার (Kirsch et al., 2019)
সার্ভাইভাল বিশ্লেষণ ক্লাসিক পাঠ্যপুস্তক (Kleinbaum & Klein, 2012)
সর্বাধিক কভারেজ সমস্যা গবেষণা (Khuller et al., 1999)
বেয়েসিয়ান সার্ভাইভাল মডেল (Qi et al., 2023)
সম্পর্কিত সক্রিয় শিক্ষা কাজ (Vinzamuri et al., 2014; Hüttel et al., 2024)

সামগ্রিক মূল্যায়ন: এটি একটি উচ্চ-মানের মেশিন লার্নিং পেপার, যা বাজেট সীমাবদ্ধতার অধীনে সার্ভাইভাল ডেটার সক্রিয় শিক্ষা সমস্যা উদ্ভাবনীভাবে সমাধান করে। পদ্ধতি ডিজাইন চতুর, তাত্ত্বিক বিশ্লেষণ কঠোর, পরীক্ষা যাচাইকরণ সম্পূর্ণ। যদিও কিছু অনুমান সীমাবদ্ধতা রয়েছে, এটি গুরুত্বপূর্ণ ব্যবহারিক প্রয়োগের জন্য কার্যকর সমাধান প্রদান করে, উচ্চ একাডেমিক মূল্য এবং ব্যবহারিক তাৎপর্য রয়েছে।